SHARE
TWEET

Untitled

a guest Jul 16th, 2019 68 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. ###############################################################################
  2. #                               4. Classifiers                                #
  3. ###############################################################################
  4. # Create list of tuples with classifier label and classifier object
  5. classifiers = {}
  6. classifiers.update({"LDA": LinearDiscriminantAnalysis()})
  7. classifiers.update({"QDA": QuadraticDiscriminantAnalysis()})
  8. classifiers.update({"AdaBoost": AdaBoostClassifier()})
  9. classifiers.update({"Bagging": BaggingClassifier()})
  10. classifiers.update({"Extra Trees Ensemble": ExtraTreesClassifier()})
  11. classifiers.update({"Gradient Boosting": GradientBoostingClassifier()})
  12. classifiers.update({"Random Forest": RandomForestClassifier()})
  13. classifiers.update({"Ridge": RidgeClassifier()})
  14. classifiers.update({"SGD": SGDClassifier()})
  15. classifiers.update({"BNB": BernoulliNB()})
  16. classifiers.update({"GNB": GaussianNB()})
  17. classifiers.update({"KNN": KNeighborsClassifier()})
  18. classifiers.update({"MLP": MLPClassifier()})
  19. classifiers.update({"LSVC": LinearSVC()})
  20. classifiers.update({"NuSVC": NuSVC()})
  21. classifiers.update({"SVC": SVC()})
  22. classifiers.update({"DTC": DecisionTreeClassifier()})
  23. classifiers.update({"ETC": ExtraTreeClassifier()})
  24.  
  25. # Create dict of decision function labels
  26. DECISION_FUNCTIONS = {"Ridge", "SGD", "LSVC", "NuSVC", "SVC"}
  27.  
  28. # Create dict for classifiers with feature_importances_ attribute
  29. FEATURE_IMPORTANCE = {"Gradient Boosting", "Extra Trees Ensemble", "Random Forest"}
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top