ium_434760/MLProject
2021-05-24 00:52:24 +02:00

11 lines
239 B
Plaintext

name: fifa_players
docker_env:
image: adnovac/ium_s434760
entry_points:
train:
parameters:
batch_size: {type: int, default: 15}
epochs: {type: int, default: 16}
command: "python3.8 train.py {batch_size} {epochs}"