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)