jasmine.linear_model.LinearRegression

class jasmine.linear_model.LinearRegression(use_bias=True, learning_rate=0.01, n_epochs=1000, loss_function=<PjitFunction of <function mse_loss>>, l1_penalty=0.0, l2_penalty=0.0, optimizer=None)

Linear regression trained with gradient descent.

center_targets = True

log_every_epoch = True

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

loss_fn(params, X, y)

Compute the training loss for the estimator.