ium_495719/MLProject
2024-05-16 04:54:53 +02:00

14 lines
340 B
Plaintext

name: HousePriceModel
conda_env: environment.yml
entry_points:
main:
parameters:
epochs: {type: int, default: 20}
learning_rate: {type: float, default: 0.001}
batch_size: {type: int, default: 32}
command: "python create_model.py {epochs} {learning_rate} {batch_size}"
test:
command: "python evaluate.py"