mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-15 11:58:07 +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
@@ -52,6 +52,12 @@ class EvenOddStubModel(Model):
|
||||
|
||||
def clone(self):
|
||||
return self
|
||||
|
||||
def get_name(self) -> str:
|
||||
return 'test'
|
||||
|
||||
def initialize_network(self, input_dim: int, output_dim: int):
|
||||
pass
|
||||
|
||||
|
||||
def test_evaluation():
|
||||
|
||||
@@ -51,6 +51,12 @@ class IncrementingStubModel(Model):
|
||||
def clone(self):
|
||||
return self
|
||||
|
||||
def get_name(self) -> str:
|
||||
return 'test'
|
||||
|
||||
def initialize_network(self, input_dim: int, output_dim: int):
|
||||
pass
|
||||
|
||||
def test_walk_forward_train_test():
|
||||
X, y = __generate_incremental_test_data(no_of_rows)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user