ium_444452/MLproject

24 lines
957 B
Plaintext
Raw Normal View History

2022-05-15 22:20:18 +02:00
name: s444452_fake_job_classification
docker_env:
image: s444452/ium:1.4
2022-05-15 22:42:16 +02:00
volumes: ["/tmp/mlruns:/mlruns:rw"]
2022-05-15 22:20:18 +02:00
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}"