Skip to content

Instantly share code, notes, and snippets.

@sean-mcclure
Created November 7, 2021 22:32
Show Gist options
  • Save sean-mcclure/a442ff12c2b5341499b3b3b0e7f1a18f to your computer and use it in GitHub Desktop.
Save sean-mcclure/a442ff12c2b5341499b3b3b0e7f1a18f to your computer and use it in GitHub Desktop.
classifier_config_dict
classifier_config_dict = {
# Classifiers
'sklearn.naive_bayes.GaussianNB': {
},
'sklearn.naive_bayes.BernoulliNB': {
'alpha': [1e-3, 1e-2, 1e-1, 1., 10., 100.],
'fit_prior': [True, False]
},
'sklearn.naive_bayes.MultinomialNB': {
'alpha': [1e-3, 1e-2, 1e-1, 1., 10., 100.],
'fit_prior': [True, False]
},
'sklearn.tree.DecisionTreeClassifier': {
'criterion': ["gini", "entropy"],
'max_depth': range(1, 11),
'min_samples_split': range(2, 21),
'min_samples_leaf': range(1, 21)
},
...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment