2024-05-16 03:01:35 +02:00
|
|
|
name: HousePriceModel
|
|
|
|
|
|
2024-05-16 04:54:53 +02:00
|
|
|
conda_env: environment.yml
|
2024-05-16 03:01:35 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: int, default: 20}
|
|
|
|
|
learning_rate: {type: float, default: 0.001}
|
2024-05-16 04:54:53 +02:00
|
|
|
batch_size: {type: int, default: 32}
|
|
|
|
|
command: "python create_model.py {epochs} {learning_rate} {batch_size}"
|
2024-05-16 03:01:35 +02:00
|
|
|
|
|
|
|
|
test:
|
2024-05-16 04:54:53 +02:00
|
|
|
command: "python evaluate.py"
|
