-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathparams.yaml
36 lines (28 loc) · 922 Bytes
/
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
external_data_config:
external_data_csv: data/external/train.csv
raw_data_config:
raw_data_csv: data/raw/train.csv
model_var: ['Temperature','Oxygen','Humidity', 'Prediction']
num_features: ['Temperature','Oxygen','Humidity']
train_test_split_ratio: 0.2
target: Prediction
random_state: 111
new_train_data_csv: data/raw/train_new.csv
processed_data_config:
train_data_csv: data/processed/forest_fire_train.csv
test_data_csv: data/processed/forest_fire_test.csv
mlflow_config:
artifacts_dir: artifacts
experiment_name: model_iteration1
run_name: random_forest
registered_model_name: random_forest_model
remote_server_uri: "http://localhost:1234"
experiment_ids: "1"
random_forest:
max_depth: 20
n_estimators: 30
model_dir:
model_webapp_dir: models/model.joblib
model_monitor:
target_col_name: target
monitor_dashboard_html: reports/data_and_target_drift_dashboard.html