diff --git a/src/mlpack/bindings/python/print_doc_functions_impl.hpp b/src/mlpack/bindings/python/print_doc_functions_impl.hpp index 4356e649dc..ba833e49d8 100644 --- a/src/mlpack/bindings/python/print_doc_functions_impl.hpp +++ b/src/mlpack/bindings/python/print_doc_functions_impl.hpp @@ -413,12 +413,13 @@ inline std::string SplitTrainTest(const std::string& datasetName, const std::string& testLabels, const std::string& splitRatio) { - std::string splitString = ">>> "; - splitString += testDataset + ", " + testLabels + ", "; - splitString += trainDataset + ", " + trainLabels; - splitString += " = "; - splitString += "preprocess_split(input_=" + datasetName + ", input_labels="; - splitString += labelName + ", test_ratio=" + splitRatio + ")"; + std::string splitString; + splitString += ">>> d = preprocess_split(input_=" + datasetName + ", input_labels="; + splitString += labelName + ", test_ratio=" + splitRatio + ")\n"; + splitString += ">>> " + trainDataset + " = d['training']\n"; + splitString += ">>> " + trainLabels + " = d['training_labels']\n"; + splitString += ">>> " + testDataset + " = d['test']\n"; + splitString += ">>> " + testLabels + " = d['test_labels']"; return splitString; } diff --git a/src/mlpack/methods/linear_regression/linear_regression_train_main.cpp b/src/mlpack/methods/linear_regression/linear_regression_train_main.cpp index 161f131b8d..9d782e0868 100644 --- a/src/mlpack/methods/linear_regression/linear_regression_train_main.cpp +++ b/src/mlpack/methods/linear_regression/linear_regression_train_main.cpp @@ -46,14 +46,14 @@ BINDING_EXAMPLE( GET_DATASET("X", "https://example.com") + "\n" + GET_DATASET("y", "https://example.com") + "\n" + SPLIT_TRAIN_TEST("X", "y", "X_train", "y_train", "X_test", "y_test", - "0.2") + "\n" + - CREATE_OBJECT("model", "linear_regression") + "\n" + - CALL_METHOD("model", "train", "training", "X_train", - "training_responses", "y_train")); + "0.2") + "\n" + + CREATE_OBJECT("lr", "linear_regression") + "\n" + + CALL_METHOD("lr", "train", "training", "X_train", "training_responses", + "y_train")); // See also... BINDING_SEE_ALSO("Linear/ridge regression tutorial", - "@doc/tutorials/linear_regression.md"); + "@doc/tutorials/linear_regression.md"); PARAM_MATRIX_IN_REQ("training", "Matrix containing training set X (regressors).", "t");