yng*_*tgy 3 python machine-learning cross-validation
我正在尝试将文本数据分类为多个类。我想执行交叉验证来比较多个模型与样本权重。
对于每个模型,我都可以输入这样的参数。
all_together = y_train.to_numpy()
unique_classes = np.unique(all_together)
c_w = class_weight.compute_class_weight('balanced', unique_classes, all_together)
clf = MultinomialNB().fit(X_train_tfidf, y_train, sample_weight=[c_w[i] for i in all_together])
Run Code Online (Sandbox Code Playgroud)
似乎不允许cross_val_score()
有关sample_weight的参数。我如何通过交叉验证来做到这一点?
models = [
RandomForestClassifier(n_estimators=200, max_depth=3, random_state=0),
LinearSVC(),
MultinomialNB(),
LogisticRegression(random_state=0),
]
all_together = y_train.to_numpy()
unique_classes = np.unique(all_together)
c_w = class_weight.compute_class_weight('balanced', unique_classes, all_together)
CV = 5
cv_df = pd.DataFrame(index=range(CV * len(models)))
entries = []
for model in models:
model_name = model.__class__.__name__
f1_micros = cross_val_score(model, X_tfidf, y_train, scoring='f1_micro', cv=CV)
for fold_idx, f1_micro in enumerate(f1_micros):
entries.append((model_name, fold_idx, f1_micro))
cv_df_women = pd.DataFrame(entries, columns=['model_name', 'fold_idx', 'f1_micro'])
Run Code Online (Sandbox Code Playgroud)
cross_val_score
有一个名为的参数fit_params
,它接受参数(键)和值的字典以传递给fit()
估计器的方法。对于你的情况,你可以这样做
cross_val_score(model, X_tfidf, y_train, scoring='f1_micro', cv=CV, fit_params={'sample_weight': [c_w[i] for i in all_together]})
Run Code Online (Sandbox Code Playgroud)
归档时间: |
|
查看次数: |
1807 次 |
最近记录: |