Notice: This page requires JavaScript to function properly.
Please enable JavaScript in your browser settings or update your browser.
Leer Train-Test Split | Identifying Spam Emails
Identifying Spam Emails

book
Train-Test Split

The train-test split is a method used in machine learning to divide a dataset into two parts: a training set and a test set.

The training set is used to train a model, while the test set is used to evaluate the model's performance. This split is crucial as it allows the model to be tested on unseen data, helping to prevent overfitting.

Overfitting occurs when a model learns the training data too well, performing poorly on unseen data. Evaluating the model on a test set provides a better indication of how it will perform in real-world scenarios.

Additionally, this approach helps to understand the model's generalization ability and allows for the tuning of hyperparameters by comparing performance across different test sets.

Taak

Swipe to start coding

  1. Import the train_test_split() function.
  2. Use this function to split the newly created X and y variables into train and test sets.

Oplossing

# Import the train_test_split() function
from sklearn.model_selection import train_test_split

# Split the data into train and test sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20, random_state=11)
print(f"Train Data Shape: {X_train.shape}\nTest Data Shape: {X_test.shape}")

Mark tasks as Completed
Was alles duidelijk?

Hoe kunnen we het verbeteren?

Bedankt voor je feedback!

Sectie 1. Hoofdstuk 8
AVAILABLE TO ULTIMATE ONLY
some-alt