【发布时间】:2018-11-15 16:16:01
【问题描述】:
我是 NLP 新手,我正在尝试构建一个文本分类器,但我的数据目前不平衡。最高类别有多达 280 个条目,而最低类别有 30 个条目。 我正在尝试对当前数据使用交叉验证技术,但是在寻找了几天之后我无法实现它。它看起来很简单,但我仍然无法实现它。这是我的代码
y = resample.Subsystem
X = resample['new description']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.20, random_state=42)
from sklearn.feature_extraction.text import CountVectorizer
count_vect = CountVectorizer()
X_train_counts = count_vect.fit_transform(X_train)
X_train_counts.shape
from sklearn.feature_extraction.text import TfidfTransformer
tfidf_transformer = TfidfTransformer()
X_train_tfidf = tfidf_transformer.fit_transform(X_train_counts)
X_train_tfidf.shape
#SVM
from sklearn.pipeline import Pipeline
from sklearn.linear_model import SGDClassifier
text_clf_svm = Pipeline([('vect', CountVectorizer(stop_words='english')),('tfidf', TfidfTransformer()),('clf-svm', SGDClassifier(loss='hinge', penalty='l2',alpha=1e-3, n_iter=5, random_state=42)),])
text_clf_svm.fit(X_train, y_train)
predicted_svm = text_clf_svm.predict(X_test)
print('The best accuracy is : ',np.mean(predicted_svm == y_test))
我已经进一步做了一些 gridsearch 和 Stemmer,但现在我将对此代码进行交叉验证。我已经很好地清理了数据,但我仍然获得 60% 的准确度 任何帮助将不胜感激
【问题讨论】:
标签: python-3.x machine-learning svm pipeline cross-validation