AdamOsiowy123
f790b4d925
Some checks failed
s444452-training/pipeline/head There was a failure building this commit
24 lines
957 B
Plaintext
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}" |