Skip to content

Instantly share code, notes, and snippets.

@Damian89

Damian89/example_tfidf.py

Created October 23, 2017 14:22
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Damian89/96c1b10b8d01f1903c86efe8acacfa15 to your computer and use it in GitHub Desktop.
Save Damian89/96c1b10b8d01f1903c86efe8acacfa15 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
example_tfidf.py
import warnings
from sklearn import ensemble
warnings.filterwarnings("ignore")
features = [
[1, 1, 0],
[1, 1, 0],
[1, 1, 0],
[0, 1, 0],
[0, 1, 0],
[0, 1, 0],
]
targets = [
1,
1,
1,
0,
0,
0,
]
model = ensemble.ExtraTreesClassifier()
model.fit(features, targets)
print(model.feature_importances_)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment