mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-17 21:08:08 +00:00
feature(Models): Implemented a basic Neural Network with Pytorch-Lightning (#101)
* feat: Added base functions for Neural Net. * feat: Added function to handle Neural Nets. * fix: Fixed fit loop * feat: Neural Net trains now, need to test it. * feat: Prediction now works on the neural net. * fix: Put back config and run_pipeline.py * fix: Took out import from run_pipeline. * fix(Models): added get_name(), adjusted pytorch model output size * fix(Tests): fixed tests Co-authored-by: Mark Aron Szulyovszky <mark.szulyovszky@gmail.com>
This commit is contained in:
co-authored by
Mark Aron Szulyovszky
parent
1cd0119589
commit
ee35332f58
+2
-2
@@ -132,5 +132,5 @@ lightning/lightning_logs/
|
||||
results.csv
|
||||
predictions.csv
|
||||
wandb/
|
||||
|
||||
.cachedir/**
|
||||
lightning_logs/
|
||||
.cachedir/**
|
||||
|
||||
Reference in New Issue
Block a user