Update documentation for creating a RegressionModel
@@ -14,9 +14,9 @@ import io.improbable.keanu.vertices.dbl.probabilistic.GaussianVertex;
* A general linear regression model that can be fitted to input and output training data.
* You can construct a regression model as follows:
* <pre>
- * LinearRegressionModel model = LinearRegressionModel.builder()
- * .setInputTrainingData(inputTrainingData)
- * .setOutputTrainingData(outputTrainingData)
+ * RegressionModel model = RegressionModel
+ * .withTrainingData(inputTrainingData, outputTrainingData)
+ * .withRegularization(RegressionRegularization.RIDGE)
* .build();
* </pre>
*/
