getml.predictors.LinearRegression
dataclass
Bases: _Predictor
Simple predictor for regression problems.
Learns a simple linear relationship using ordinary least squares (OLS) regression:
The weights are optimized by minimizing the squared loss of the predictions \(\hat{y}\) w.r.t. the target \(y\).
Linear regressions can be trained arithmetically or numerically. Training arithmetically is more accurate, but suffers worse scalability.
If you decide to pass categorical features to the
LinearRegression
, it will be trained
numerically. Otherwise, it will be trained arithmetically.
PARAMETER | DESCRIPTION |
---|---|
learning_rate |
The learning rate used for training numerically (only relevant when categorical features are included). Range: (0, ∞]
TYPE:
|
reg_lambda |
L2 regularization parameter. Range: [0, ∞]
TYPE:
|
validate
Checks both the types and the values of all instance variables and raises an exception if something is off.
PARAMETER | DESCRIPTION |
---|---|
params |
A dictionary containing the parameters to validate. If nothing is passed, the default parameters will be validated. |
Example
l = getml.predictors.LinearRegression()
l.learning_rate = 8.1
l.validate()
Note
This method is called at end of the __init__
constructor
and every time before the predictor - or a class holding
it as an instance variable - is sent to the getML Engine.
Source code in getml/predictors/linear_regression.py
73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
|