seldonian.models.models.BaseLogisticRegressionModel¶
- class BaseLogisticRegressionModel¶
Bases:
ClassificationModel
- __init__()¶
Base class for binary and multi-class logistic regression
- __repr__()¶
Return repr(self).
Methods
- fit(X, Y)¶
Train the model using features and labels. Let:
i = number of datapoints j = number of features (including bias term, if provied) k = number of classes
- Parameters:
X (array of shape (i,j)) – The features
Y (array of shape (i,k)) – The labels
- Returns:
fitted model weights
- Return type:
array of shape (j,k)