From 77ea0c1ad040e7fc4772c9c577d936a21ef0f722 Mon Sep 17 00:00:00 2001 From: amirmoghi3 Date: Thu, 31 Dec 2020 22:25:34 +0330 Subject: [PATCH] fix --- app.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/app.py b/app.py index dadd4e6..fffd376 100644 --- a/app.py +++ b/app.py @@ -8,7 +8,7 @@ from sklearn.metrics import classification_report # Function importing Dataset def importdata(): - dataset = pd.read_csv( 'Train.csv',sep= ',', header = None) + dataset = pd.read_csv('Train.csv',sep= ',', header = None) return dataset # Function to split the dataset @@ -16,7 +16,7 @@ def splitdataset(dataset): # Separating the target variable X = dataset.values[:, 1:5] - Y = dataset.values[:, 0] + Y = dataset.values[:,-1] # Splitting the dataset into train and test X_train, X_test, y_train, y_test = train_test_split( @@ -74,7 +74,8 @@ def main(): # Building Phase 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_entropy = tarin_using_entropy(X_train, X_test, y_train)