这是结合使用DecisionTreeClassifier,tfidf矢量化器和onevsrest分类器的正确方法吗?

时间:2019-11-19 13:03:42

标签: decision-tree tfidfvectorizer

我正在结合使用DecisionTreeClassifier,tfidf矢量化器和onevsrest分类器。 这是代码-

tfidf_vectorizer = TfidfVectorizer(max_df=0.8, max_features=800)
xtrain_tfidf = tfidf_vectorizer.fit_transform(xtrain)
xval_tfidf = tfidf_vectorizer.transform(xval)
dt=DecisionTreeClassifier()
clf = OneVsRestClassifier(dt)
clf.fit(xtrain_tfidf, ytrain)
y_pred = clf.predict(xval_tfidf)

这是正确的方法吗?

0 个答案:

没有答案