sayyedAhmed's picture
Update README.md
dc0b924 verified
metadata
license: mit
language:
  - en
metrics:
  - accuracy
pipeline_tag: time-series-forecasting
library_name: transformers
tags:
  - lstm
  - regression
  - transformer
  - inform_Severity_prediction
  - crisis_prediction

The imputation and model pipelines for the INFORM Severity data.

Ridge Regression with 88% accuracy(alpha=25 and 10-fold CV)

LightGBM with 99% accuracy (10 fold CV and reg_alpha: 1.96248962962756, reg_lambda: 4.688549449064171)

There is a 1 single pipeline for FeatureEngineering, Categorical encoding, Transformations and Imputations.

We just need to deploy the final_pipelines for both the models stored in file: final_pipe.pkl