Deploying-ML-Model-Heroku-FastAPI/config.yml at main · desared/Deploying-ML-Model-Heroku-FastAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
main:
  project_name:  Deploying a Machine Learning Model on Heroku with FastAPI
  experiment_name: development
  steps: all

data:
  cat_features : [
      "workclass",
      "education",
      "marital-status",
      "occupation",
      "relationship",
      "race",
      "sex",
      "native-country",
  ]
infer:
  update_keys:
    [
      ["marital-status","marital_status"],
      ["hours-per-week","hoursPerWeek"],
      ["native-country","nativeCountry"]
    ]
  columns:
    [ 'age',
      'workclass',
      'fnlgt',
      'education',
      'marital-status',
      'occupation',
      'relationship',
      'race',
      'sex',
      'hours-per-week',
      'native-country'
    ]