drift

mirror of https://github.com/webclinic017/drift.git synced 2026-07-28 03:08:01 +00:00

Files

T

Daniel Szemerey 55f083638f feature(Inference): Created the inference process, added model saving. (#153 )

* feat: Basic scaffolding up for inference process after training.

* feat: Saving and loading models works. Inference works nearly.

* feat: Added inference pipeline.

* feat: Saving model now accoring to date and time; loading models now selects from latest file. Fixed the creation of dictionary of models.

* feat: Added lightweight asset config, but full pipeline.

* feat: Added new naming for dictionary.

* fix: Fixed dictionary naming convention.

* fix: Fixed naming again, now the model structure is good

* fix: Changed the output path and the return values from run_pipeline.

* feat: Added function to make sure folder exists for output models.

Co-authored-by: Daniel Szemerey <szemereydaniel@gmail.com>

2022-01-11 19:15:58 +01:00

pytorch

feat(Models): added lightGBM, moved other models to separate files (#128 )

2022-01-08 12:02:42 +01:00

average.py

feature(Models): Implemented a basic Neural Network with Pytorch-Lightning (#101 )

2022-01-05 12:25:03 +01:00

base.py

feat(Models): added lightGBM, moved other models to separate files (#128 )

2022-01-08 12:02:42 +01:00

model_map.py

feat(Sweep): try to filter out some not great models (#144 )