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

refactor(Types): added nested types for Reporting (#162)

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2022-01-13 09:21:07 +01:00
committed by GitHub
parent 1856fcad22
commit 3c2a0d4247
8 changed files with 73 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/meta_labeling.py
+2 -2
View File
@@ -5,7 +5,7 @@ from feature_selection.feature_selection import select_features
import pandas as pd
from models.model_map import default_feature_selector_regression, default_feature_selector_classification
from models.base import Model
from utils.encapsulation import Single_Model
from reporting.types import Reporting
def train_meta_labeling_model(
@@ -19,7 +19,7 @@ def train_meta_labeling_model(
model_config: dict,
training_config: dict,
model_suffix: str
) -> tuple[pd.Series, pd.Series, pd.DataFrame, list[Single_Model]]:
) -> tuple[pd.Series, pd.Series, pd.DataFrame, list[Reporting.Single_Model]]:
discretize = discretize_threeway_threshold(0.33)