Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

params.yaml 1.2 KB

You have to be logged in to leave a comment. Sign In
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
  1. modeling:
  2. n_trials: 5
  3. model_family: logistic_regression # gradient_boosting #random_forest # , logistic_regression, hist_gradient_boosting, lightgbm
  4. objective_function: Logloss # Logloss, CrossEntropy
  5. loss_function: F1 # Precision, F1, Accuracy
  6. data:
  7. raw_data_path: data/raw/Weather-related disease prediction.csv
  8. interim_data_path: data/interim/weather_disease.parquet
  9. x_train_path: data/processed/x_train.csv
  10. y_train_path: data/processed/y_train.csv
  11. x_test_path: data/processed/x_test.csv
  12. y_test_path: data/processed/y_test.csv
  13. output_dir: data/processed
  14. target_col: prognosis
  15. directories:
  16. model_dir: models
  17. processed_data_dir: data/processed
  18. reports_dir: reports
  19. eda:
  20. output_prefix: weather-health-eda
  21. artifacts:
  22. model_path: models/weather_disease_model.pkl
  23. scaler_path: data/processed/minmax_scaler.pkl
  24. label_encoder_path: data/processed/label_encoder.pkl
  25. reports:
  26. model_metrics_path: reports/model_metrics.csv
  27. predictions_path: reports/predictions.csv
  28. feature_importance_path: reports/feature_importance.png
  29. confusion_matrix_path: reports/confusion_matrix.png
  30. classification_report_path: reports/classification_report.txt
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...