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}"