Coverage for mindsdb / integrations / handlers / anomaly_detection_handler / utils.py: 0%

1from pyod.models.ecod import ECOD # unsupervised default 

2from pyod.models.xgbod import XGBOD # semi-supervised default 

3from catboost import CatBoostClassifier # supervised default 

4 

5 

6def train_unsupervised(X_train, model=None): 

7 model = model if model is not None else ECOD() 

8 model.fit(X_train) 

9 return model 

10 

11 

12def train_semisupervised(X_train, y_train): 

13 model = XGBOD(estimator_list=[ECOD()]) 

14 model.fit(X_train, y_train) 

15 return model 

16 

17 

18def train_supervised(X_train, y_train, model=None): 

19 model = model if model is not None else CatBoostClassifier(logging_level="Silent") 

20 model.fit(X_train, y_train) 

21 return model