Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- model = Pipeline([('vectorizer', CountVectorizer(ngram_range=(1,2))),
- ('tfidf', TfidfTransformer(use_idf=True)),
- ('clf', OneVsRestClassifier(LinearSVC(class_weight="balanced")))])
- model.fit(list(train_data.text), list(train_data.label))
Add Comment
Please, Sign In to add comment
