iumKC/MLProject
2024-05-13 18:51:43 +02:00

11 lines
250 B
Plaintext

name: iumKC
conda_env: environment.yaml
entry_points:
main:
command: "python train.py --train=true --save_model=true"
test:
parameters:
model_filepath: "model"
command: "python test.py --test=true --load_model=$model_filepath"