-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathparams.yaml
53 lines (42 loc) · 1.16 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
41
42
43
44
45
46
47
48
49
50
51
external_data_config:
external_data_csv: data/external/train.csv
raw_data_config:
raw_data_csv: data/raw/train.csv
model_var:
[
"churn",
"number_vmail_messages",
"total_day_calls",
"total_eve_minutes",
"total_eve_charge",
"total_intl_minutes",
"number_customer_service_calls",
]
train_test_split_ratio: 0.2
target: churn
random_state: 111
new_train_data_csv: data/raw/train.csv
processed_data_config:
train_data_csv: data/processed/churn_train.csv
test_data_csv: data/processed/churn_test.csv
# params of the model
random_forest:
max_depth: 100
n_estimators: 40
mlflow_config:
artifacts_dir: artifacts
experiment_name: model_iteration1
run_name: random_forest
registered_model_name: random_forest_model
remote_server_uri: http://localhost:4000
model_dir: models/model.joblib
model_webapp_dir: models/model.joblib
model_monitor:
target_col_name: target
monitor_dashboard_html: reports/data_and_target_drift_dashboard.html
reports:
params: report/params.json
metrics: report/metrics.json
best_model_reports:
params: report/best/best_params.json
metrics: report/best/best_metrics.json