apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix call to preprocess_split() in generated documentation.

Browse Source
This commit is contained in:
Ryan Curtin
2023-11-24 12:26:58 -05:00
parent 751ba46521
commit ca8ffcd796
2 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/bindings/python/print_doc_functions_impl.hpp
+7 -6
View File
@@ -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;
}
src/mlpack/methods/linear_regression/linear_regression_train_main.cpp
+5 -5
View File
@@ -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");