ium_434749/MLProject

25 lines
854 B
Plaintext
Raw Permalink Normal View History

2021-05-23 17:00:43 +02:00
name: cnn
conda_env: conda_env.yaml
# Can have a docker_env instead of a conda_env, e.g.
# docker_env:
# image: mlflow-docker-example
entry_points:
main:
parameters:
data_file: path
regularization: {type: float, default: 0.1}
batch_size: {type: int, default: 32}
learning_rate: {type: float, default: 0.001}
epochs: {type: int, default: 2}
command: "python train_model.py with 'batch_size={batch_size}' 'learning_rate=${learning_rate}' 'epochs=${epochs}'"
validate:
parameters:
data_file: path
regularization: {type: float, default: 0.1}
batch_size: {type: int, default: 32}
learning_rate: {type: float, default: 0.001}
epochs: {type: int, default: 2}
command: "python train_model.py with 'batch_size={batch_size}' 'learning_rate=${learning_rate}' 'epochs=${epochs}'"