diff --git a/README.md b/README.md index 37dc186..3cb5f47 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,8 @@ # Plotabit Dataset link: https://www.kaggle.com/datasets/fedesoriano/stellar-classification-dataset-sdss17 + Deep Learning: https://machinelearningmastery.com/tutorial-first-neural-network-python-keras/ + Python Machine Learning: https://machinelearningmastery.com/machine-learning-in-python-step-by-step/ ## Columns diff --git a/src/main.py b/src/main.py index 947baa4..0f89d00 100755 --- a/src/main.py +++ b/src/main.py @@ -56,7 +56,7 @@ def model_switch(choice): elif (choice == 2): model = DecisionTreeClassifier(random_state=0, max_depth=20) elif (choice == 3): - model = RandomForestClassifier(n_estimators=100 ,criterion='entropy') + model = RandomForestClassifier(n_estimators=100 ,criterion='entropy', n_jobs=-1) elif (choice == 4): model = SGDClassifier(max_iter=1000, tol=0.01) elif (choice == 5): @@ -112,7 +112,7 @@ def training(model, x, y): ypredit = model.predict(Xtest) os.system("clear") - res = int(input()) + res = -1 while(res != 0): print(" Rentre un chiffre:\n\n1 - Stats %\n2 - Stats raw\n3 - accuracy_score") print("0 - QUIT")