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); }