public class SVM extends Object implements Serializable
Constructor and Description |
---|
SVM() |
Modifier and Type | Method and Description |
---|---|
org.apache.spark.api.java.JavaRDD<scala.Tuple2<Object,Object>> |
test(org.apache.spark.mllib.classification.SVMModel svmModel,
org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> testingDataset)
This method performs a binary classification using a given SVM model and a dataset.
|
org.apache.spark.mllib.classification.SVMModel |
train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset,
int noOfIterations,
double regularizationParameter)
SVM train - overload method with 2 parameters.
|
org.apache.spark.mllib.classification.SVMModel |
train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset,
int noOfIterations,
String regularizationType,
double regularizationParameter,
double initialLearningRate,
double miniBatchFraction)
This method uses stochastic gradient descent (SGD) algorithm to train a support vector machine (SVM) model.
|
public org.apache.spark.mllib.classification.SVMModel train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset, int noOfIterations, String regularizationType, double regularizationParameter, double initialLearningRate, double miniBatchFraction)
trainingDataset
- Training dataset as a JavaRDD of LabeledPointsnoOfIterations
- Number of iterarationsregularizationType
- Regularization type: L1 or L2regularizationParameter
- Regularization parameterinitialLearningRate
- Initial learning rate (SGD step size)miniBatchFraction
- SGD minibatch fractionpublic org.apache.spark.mllib.classification.SVMModel train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset, int noOfIterations, double regularizationParameter)
trainingDataset
- Training dataset as a JavaRDD of LabeledPointsnoOfIterations
- Number of iterarationsregularizationParameter
- Regularization parameterpublic org.apache.spark.api.java.JavaRDD<scala.Tuple2<Object,Object>> test(org.apache.spark.mllib.classification.SVMModel svmModel, org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> testingDataset)
svmModel
- SVM modeltestingDataset
- Testing dataset as a JavaRDD of LabeledPointsCopyright © 2015 WSO2, Inc.. All Rights Reserved.