Skip to content

Instantly share code, notes, and snippets.

Created Jul 10, 2021
What would you like to do?
import numpy
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
data = pd.read_csv("")
x = np.array(data[["Age", "EstimatedSalary"]])
y = np.array(data[["Purchased"]])
xtrain, xtest, ytrain, ytest = train_test_split(x, y, test_size=0.10, random_state=42)
model = DecisionTreeClassifier(), ytrain)
predictions = model.predict(xtest)
# Calculation of F-beta Score
from sklearn.metrics import fbeta_score
print(fbeta_score(ytest, predictions, beta=1))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment