ium_444452/MLproject
AdamOsiowy123 f790b4d925
Some checks failed
s444452-training/pipeline/head There was a failure building this commit
mlflow fix
2022-05-15 22:42:16 +02:00

24 lines
957 B
Plaintext

name: s444452_fake_job_classification
docker_env:
image: s444452/ium:1.4
volumes: ["/tmp/mlruns:/mlruns:rw"]
entry_points:
train:
parameters:
data_path: path
epochs: {type: float, default: 1}
num_words: {type: float, default: 20000}
batch_size: {type: float, default: 150}
pad_length: {type: float, default: 300}
command: "python3 Scripts/train_neural_network.py {data_path} {epochs} {num_words} {batch_size} {pad_length}"
evaluate:
parameters:
build_nr: {type: float, default: 0}
data_path: path
epochs: {type: float, default: 1}
num_words: {type: float, default: 20000}
batch_size: {type: float, default: 150}
pad_length: {type: float, default: 300}
command: "python3 Scripts/evaluate_neural_network.py {build_nr} {data_path} {epochs} {num_words} {batch_size} {pad_length}"