sparkRegressor.py
Fits a regression model using an MLlib regression method and returns regression metrics
SparkRegressor(predictor, label, testFraction=0.3, seed=1)[source]¶Bases: object
Fits a regression model using an MLlib regression method and returns regression metrics
Attributes
| predictor | type of multi-class classifier |
| label | (str) classification label |
| testFraction | (float) test set fraction [0.3] |
| seed | (int) random seed |
Methods
fit(data) |
Dataset must at least contain the following two columns: label : the class labels features : feature vector |