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

chore(Linter): reformatted code with black (#211)

Browse Source 
* chore(Linter): reformatted code with black

* Create black.yaml
This commit is contained in:
Mark Aron Szulyovszky
2022-02-17 19:22:17 +01:00
committed by GitHub
parent f3fee4a4e1
commit 8dd2d88740
101 changed files with 2595 additions and 2319 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/sklearn.py
+2 -6
View File
@@ -4,20 +4,16 @@ from .base import Model
import numpy as np
def SKLearnModel(instance) -> Model:
instance.data_transformation = 'transformed'
instance.data_transformation = "transformed"
instance.only_column = None
instance.predict_window_size = 'single_timestamp'
instance.predict_window_size = "single_timestamp"
instance.name = instance.__class__.__name__
return instance
# def predict(self, X) -> tuple[float, np.ndarray]:
# pred = self.model.predict(X).item()
# probability = self.model.predict_proba(X).squeeze()
# return (pred, probability)