public class SupervisedSparkModelBuilder extends MLModelBuilder
Constructor and Description |
---|
SupervisedSparkModelBuilder(MLModelConfigurationContext context) |
Modifier and Type | Method and Description |
---|---|
protected long |
arraySum(double[] array)
This summation of a given double array
|
org.wso2.carbon.ml.commons.domain.MLModel |
build()
Build a supervised model.
|
protected Double |
getModelAccuracy(org.apache.spark.mllib.evaluation.MulticlassMetrics multiclassMetrics)
This method gets model accuracy from given multi-class metrics
|
protected MulticlassConfusionMatrix |
getMulticlassConfusionMatrix(org.apache.spark.mllib.evaluation.MulticlassMetrics multiclassMetrics,
org.wso2.carbon.ml.commons.domain.MLModel mlModel)
This method returns multiclass confusion matrix for a given multiclass metric object
|
protected org.apache.spark.mllib.evaluation.MulticlassMetrics |
getMulticlassMetrics(org.apache.spark.api.java.JavaSparkContext sparkContext,
org.apache.spark.api.java.JavaPairRDD<Double,Double> predictionsAndLabels)
This method gets multi class metrics for a given set of prediction and label values
|
org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> |
preProcess() |
getContext, handleIgnoreException, setContext
public SupervisedSparkModelBuilder(MLModelConfigurationContext context)
public org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> preProcess() throws MLModelBuilderException
MLModelBuilderException
public org.wso2.carbon.ml.commons.domain.MLModel build() throws MLModelBuilderException
build
in class MLModelBuilder
MLModel
MLModelBuilderException
- if failed to build the model.protected org.apache.spark.mllib.evaluation.MulticlassMetrics getMulticlassMetrics(org.apache.spark.api.java.JavaSparkContext sparkContext, org.apache.spark.api.java.JavaPairRDD<Double,Double> predictionsAndLabels)
sparkContext
- JavaSparkContextpredictionsAndLabels
- Prediction and label values RDDprotected MulticlassConfusionMatrix getMulticlassConfusionMatrix(org.apache.spark.mllib.evaluation.MulticlassMetrics multiclassMetrics, org.wso2.carbon.ml.commons.domain.MLModel mlModel)
multiclassMetrics
- Multiclass metric objectprotected Double getModelAccuracy(org.apache.spark.mllib.evaluation.MulticlassMetrics multiclassMetrics)
multiclassMetrics
- multi-class metrics objectprotected long arraySum(double[] array)
array
- Double arrayCopyright © 2016 WSO2, Inc.. All Rights Reserved.