ium_434760/MLProject

11 lines
296 B
Plaintext
Raw Normal View History

2021-05-16 23:41:55 +02:00
name: Fifa Players
docker_env:
image: docker.io/adnovac/ium_s434760:2.0
entry_points:
train:
parameters:
batch_size: {type: int, default: 15}
epochs: {type: int, default: 16}
command: "python train.py {batch_size} {epochs}"
evaluate:
command: "python evaluate.py"