jasmine.linear_model.LogisticRegression

class jasmine.linear_model.LogisticRegression(use_bias=True, learning_rate=0.01, n_epochs=1000, loss_function=<function bce_loss>, l1_penalty=0.0, l2_penalty=0.0, optimizer=None)

Binary logistic regression trained with gradient descent.

stop_on_invalid_loss = True

__init__(use_bias=True, learning_rate=0.01, n_epochs=1000, loss_function=<function bce_loss>, l1_penalty=0.0, l2_penalty=0.0, optimizer=None)

loss_fn(params, X, y)

Compute the training loss for the estimator.