-
Notifications
You must be signed in to change notification settings - Fork 0
/
params.yaml
40 lines (34 loc) · 1.04 KB
/
params.yaml
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
37
38
39
40
base:
project_name: cicd
target: quality
load_dataset:
bucket: datasetbucket
filename: WineQT.csv
raw_data: /data/raw
make_dataset:
test_split: 0.25
seed: 41
processed_data: /data/interim
res_seed: 42
build_features:
processed_data: /data/processed
mlflow_config:
trainingExpName: 'Trained Models'
tunningExpName: 'Tunned Models'
bestModelExpName: 'Best Models'
remote_server_uri: http://localhost:5000
mlflow_tracking_uri: mysql+pymysql://admin:[email protected]:3306/mysql_mlflow
reg_model_name: 'RFC Model'
stage: production
# cmd to activate server: mlflow server -h 127.0.0.1 -p 5000 --backend-store-uri mysql+pymysql://admin:[email protected]:3306/mysql_mlflow --default-artifact-root s3://mymlflowartifacts/cicdMLflowArtifacts
train_model:
n_estimators: 100
criterion: gini
max_depth: 18
min_samples_leaf: 30
random_state: 42
min_samples_split: 60
model_dir: /models
hyperopt:
max_eval: 10
model_name: best_model