formation-mlops/MLproject at main · parhammarstrom/formation-mlops · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
name: text-classification

entry_points:
  main:
    parameters:
      remote_server_uri: {type: str, default: https://mlflow.lab.sspcloud.fr}
      experiment_name: {type: str, default: nace-prediction}
      run_name: {type: str, default: default}
      dim: {type: int, default: 150}
      lr: {type: float, default: 0.2}
      epoch: {type: int, default: 25}
      wordNgrams: {type: int, default: 3}
      minn: {type: int, default: 3}
      maxn: {type: int, default: 4}
      minCount: {type: int, default: 3}
      bucket: {type: int, default: 100000}
      thread: {type: int, default: 10}
    command: "python src/train.py {remote_server_uri} {experiment_name} {run_name} {dim} {lr} {epoch} {wordNgrams} {minn} {maxn} {minCount} {bucket} {thread}"