-
Notifications
You must be signed in to change notification settings - Fork 0
/
params.yaml
30 lines (27 loc) · 1.15 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
# Global parameters
random_state: 42
target: "loan_status"
train_file: "data/processed/train.csv"
test_file: "data/processed/test.csv"
train_metrics_file: "reports/train_metrics.json"
test_metrics_file: "reports/test_metrics.json"
train_calibration_curve_file: "reports/figures/train_calibration_curve.png"
test_calibration_curve_file: "reports/figures/test_calibration_curve.png"
model_file: "models/woelr.pkl"
# Parameters for data preprocessing
data_preprocessing:
raw_data_file: "data/raw/credit_risk_dataset.csv"
test_size: 0.3
# Parameters about model training
train:
weight_of_evidence_transformer:
numerical_columns: ["person_age", "person_income", "person_emp_length", "loan_amnt", "loan_int_rate", "loan_percent_income", "cb_person_cred_hist_length"]
categorical_columns: ["person_home_ownership", "loan_intent", "loan_grade", "cb_person_default_on_file"]
bins: 8
logistic_regression:
random_state: 42
# Parameters containing information for experiment tracking
mlflow:
experiment_name: "weight-of-evidence-logistic-regression-exp"
model_name: "woelr"
remote_uri: "https://dagshub.com/marcellinus-witarsah/credit-score-mlops.mlflow"