komodoml.models

Models module for KomodoML.

class komodoml.models.DecisionTreeClf(**kwargs)

Bases: BaseModel

Decision Tree Classifier model.

Parameters:

**kwargs (dict) – Additional keyword arguments to pass to the DecisionTreeClassifier.

model

The underlying Decision Tree Classifier model.

Type:

DecisionTreeClassifier

class komodoml.models.DecisionTreeReg(**kwargs)

Bases: BaseModel

Decision Tree Regressor model.

Parameters:

**kwargs (dict) – Additional keyword arguments to pass to the DecisionTreeRegressor.

model

The underlying Decision Tree Regressor model.

Type:

DecisionTreeRegressor