diff --git a/Part 1 - Introduction to Machine Learning with scikit-learn.md b/Part 1 - Introduction to Machine Learning with scikit-learn.md index 91a2b23..0a9a8e8 100644 --- a/Part 1 - Introduction to Machine Learning with scikit-learn.md +++ b/Part 1 - Introduction to Machine Learning with scikit-learn.md @@ -72,9 +72,13 @@ X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target, In the above example, we import the `train_test_split` method from scikit-learn's `model_selection` sublibrary and use it to generate four smaller arrays: `X_train`, a two-dimensional array containing a certain amount of entries from the main dataset. Does not include the expected outcome of each data entry. + `Y_train`, a one-dimensional array containing the expected outcome of each data entry in `X_train`. +  + `X_test`, a two-dimensional array containing a certain amount of entries from the main dataset. Does not include the expected outcome of each data entry. + `Y_test`, a one-dimensional array containing the expected outcome of each data entry in `X_test`. Continuing our analogy of studying for a math exam,