sparkMultiClassClassifier.py
Fits a multi-class classification model using mllib classification method and returns classification metrics.
SparkMultiClassClassifier
(predictor, label, testFraction=0.3, seed=1)[source]¶Bases: object
Fits a multi-class classification model using mllib classification method and returns classification 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 |