Skip to content

Instantly share code, notes, and snippets.

@dbulgakov

dbulgakov/gist:55edd5eeccf112ad44bb10b1414ea21c

Created March 5, 2017 19:44
Show Gist options
  • Save dbulgakov/55edd5eeccf112ad44bb10b1414ea21c to your computer and use it in GitHub Desktop.
Save dbulgakov/55edd5eeccf112ad44bb10b1414ea21c to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. dbulgakov created this gist Mar 5, 2017.
    7 changes: 7 additions & 0 deletions gistfile1.txt
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,7 @@
    from sklearn.naive_bayes import GaussianNB
    cv = CountVectorizer(ngram_range=(1, 5))
    X = cv.fit_transform(X_train)
    X_t = cv.transform(X_test)
    clf = GaussianNB()
    clf.fit(X.toarray(), y_train)
    y_pred = clf.predict(X_t.toarray())