public MyJavaLogisticRegressionModel train(DataFrame dataset) {
JavaRDD<LabeledPoint> oldDataset = extractLabeledPoints(dataset).toJavaRDD();
int numFeatures = oldDataset.take(1).get(0).features().size();
Vector weights = Vectors.zeros(numFeatures); // Learning would happen here.
return new MyJavaLogisticRegressionModel(uid(), weights).setParent(this);
}
|