gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-21 06:48:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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:
Daniel Szemerey
2022-01-05 12:25:03 +01:00
committed by GitHub
co-authored by Mark Aron Szulyovszky
parent 1cd0119589
commit ee35332f58
12 changed files with 197 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
run_pipeline.py
-2
View File
@@ -21,8 +21,6 @@ def setup_pipeline(project_name:str, with_wandb: bool, sweep: bool):
model_config, training_config, data_config = preprocess_config(model_config, training_config, data_config)
pipeline(project_name, wandb, sweep, model_config, training_config, data_config)
def pipeline(project_name:str, wandb, sweep:bool, model_config:dict, training_config:dict, data_config:dict):
results = pd.DataFrame()