Course Wiki

Setting up the validation framework — Machine Learning Zoomcamp Module 3

Splitting the dataset with Scikit-Learn.

Machine Learning Zoomcamp › Module 3: Machine Learning for Classification › Setting up the validation framework

Notes

Splitting the dataset with Scikit-Learn.

Classes, functions, and methods:

• train_test_split - Scikit-Learn class for splitting a dataset into two parts. The test_size argument states how large the test set should be. The random_state argument sets a random seed for reproducibility purposes.
• df.reset_index(drop=True) - reset the indices of a dataframe and delete the previous ones.
• df.x.values - extract the values from x series
• del df['x'] - delete x series from a dataframe

The entire code of this project is available in this jupyter notebook.

• Notes from Peter Ernicke

Key concepts

• Pandas
• scikit-learn
• Jupyter Notebooks

Related notes

• mlz-m01-introduction-to-numpy
• mlz-m01-introduction-to-pandas
• mlz-m01-linear-algebra-refresher
• mlz-m01-setting-up-the-environment
• mlz-m01-summary
• mlz-m02-baseline-model-for-car-price-prediction-project
• mlz-m02-categorical-variables
• mlz-m02-computing-rmse-on-validation-data
• mlz-m02-data-preparation
• mlz-m02-exploratory-data-analysis
• mlz-m02-explore-more
• mlz-m02-feature-engineering
• mlz-m02-linear-regression
• mlz-m02-linear-regression-vector-form
• mlz-m02-regularization
• mlz-m02-root-mean-squared-error-rmse
• mlz-m02-setting-up-the-validation-framework
• mlz-m02-training-linear-regression-normal-equation
• mlz-m02-tuning-the-model
• mlz-m02-using-the-model
• mlz-m03-churn-prediction-project
• mlz-m03-data-preparation
• mlz-m03-eda
• mlz-m03-explore-more
• mlz-m03-feature-importance-churn-rate-and-risk-ratio
• mlz-m03-feature-importance-correlation
• mlz-m03-feature-importance-mutual-information
• mlz-m03-logistic-regression
• mlz-m03-model-interpretation
• mlz-m03-one-hot-encoding
• mlz-m03-summary
• mlz-m03-training-logistic-regression-with-scikit-learn
• mlz-m03-using-the-model
• mlz-m04-accuracy-and-dummy-model
• mlz-m04-confusion-table
• mlz-m04-cross-validation
• mlz-m04-roc-auc
• mlz-m04-roc-curves
• mlz-m04-summary
• mlz-m05-python-virtual-environment-pipenv
• mlz-m06-data-cleaning-and-preparation
• mlz-m06-decision-tree-learning-algorithm
• mlz-m06-decision-trees
• mlz-m06-decision-trees-parameter-tuning
• mlz-m06-ensemble-learning-and-random-forest
• mlz-m06-gradient-boosting-and-xgboost
• mlz-m06-xgboost-parameter-tuning
• mlz-m08-installation-of-tensorflow
• mlz-m09-explore-more
• mlz-m10-creating-a-pre-processing-service

Sources

• Video
• Lesson file