amirmoghi3
/
decision-tree-with-py-libs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

master
Amir Hossein Moghiseh 2020-12-31 22:25:34 +03:30
parent ea2218b219
commit 77ea0c1ad0
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app.py
View File

@ -16,7 +16,7 @@ def splitdataset(dataset):
# Separating the target variable # Separating the target variable
X = dataset.values[:, 1:5] X = dataset.values[:, 1:5]
Y = dataset.values[:, 0] Y = dataset.values[:,-1]
# Splitting the dataset into train and test # Splitting the dataset into train and test
X_train, X_test, y_train, y_test = train_test_split( X_train, X_test, y_train, y_test = train_test_split(
@ -75,6 +75,7 @@ def main():
# Building Phase # Building Phase
data = importdata() data = importdata()
X, Y, X_train, X_test, y_train, y_test = splitdataset(data) X, Y, X_train, X_test, y_train, y_test = splitdataset(data)
clf_gini = train_using_gini(X_train, X_test, y_train) clf_gini = train_using_gini(X_train, X_test, y_train)
clf_entropy = tarin_using_entropy(X_train, X_test, y_train) clf_entropy = tarin_using_entropy(X_train, X_test, y_train)