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 |
|