classifier

Module Contents

Classes

XGBClassifierModel
XGBClassifierModeler

class classifier.XGBClassifierModel

Bases: h1st.model.ml.xgboost.regression.XGBRegressionModel

name = XGBClassifier

predict(self, input_data: dict) → dict

apply_threshold(self, x)

set_threshold(self, threshold: float)

class classifier.XGBClassifierModeler(threshold=0.5, **kwargs)

Bases: h1st.model.ml.xgboost.regression.XGBRegressionModeler

model_class