ium_464863/mlflow/breast_cancer_pytorch/MLproject

11 lines
312 B
Plaintext

name: s464863
conda_env: conda.yaml
entry_points:
main:
parameters:
learning_rate: {type: float, default: 0.001}
weight_decay: {type: float, default: 0.001}
num_epochs: {type: int, default: 1000}
command: "python ../create_model.py {learning_rate} {weight_decay} {num_epochs}"