Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/lazy predict classification1.py Secret

Created June 19, 2021 12:41
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/0c595d992a2015a5db8d51eec2e6c288 to your computer and use it in GitHub Desktop.
Save amankharwal/0c595d992a2015a5db8d51eec2e6c288 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
lazy predict classification1.py
from lazypredict.Supervised import LazyClassifier
import numpy
import pandas as pd
data = pd.read_csv("https://raw.githubusercontent.com/amankharwal/Website-data/master/social.csv")
x = np.array(data[["Age", "EstimatedSalary"]])
y = np.array(data[["Purchased"]])
from sklearn.model_selection import train_test_split
xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size=0.10, random_state=42)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment