text_plain_output_1.png
from sklearn.model_selection import train_test_split

import pandas as pd

dataset = pd.read_csv('/kaggle/input/phishing-website-dataset/dataset.csv')

selected_features = ['URLURL_Length', 'having_At_Symbol', 'double_slash_redirecting', 'Prefix_Suffix', 'Page_Rank', 'Google_Index', 'Result']
df = dataset[selected_features]

df = df.drop_duplicates()
df.shape

X = df.drop('Result', axis=1)
y = df['Result']

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0, stratify=y)