python - 网格搜索错误

Question

我一直在尝试执行网格搜索，但似乎有些问题。我的代码是:

grid_search_0 = GridSearchCV(estimator=Pipeline([('vectorizer', CountVectorizer()), ('tfidf', TfidfTransformer()), ('clf', LinearSVC())]),
             param_grid={'C': 3**np.arange(-3, 3, dtype='float'),
                         'gamma': 3**np.arange(-6, 0, dtype='float'), },
             cv=10,
             scoring=make_scorer(roc_auc_score, needs_threshold=True),
             verbose=1,
             n_jobs=-1,)

我收到错误

ImportError: [joblib] 尝试在不支持 fork 的系统上进行并行计算而不保护您的导入。要在脚本中使用并行计算，必须使用“if __name__ == '__main__'”保护主循环。请参阅并行的 joblib 文档以获取更多信息


有人遇到过这个问题并解决过吗？我做错了什么？

Answer 1

这是错误消息建议执行的操作，这对您有用吗？

if __name__ == '__main__':

    grid_search_0 = GridSearchCV(estimator=Pipeline([('vectorizer', CountVectorizer()), ('tfidf', TfidfTransformer()), ('clf', LinearSVC())]),
             param_grid={'C': 3**np.arange(-3, 3, dtype='float'),
                         'gamma': 3**np.arange(-6, 0, dtype='float'), },
             cv=10,
             scoring=make_scorer(roc_auc_score, needs_threshold=True),
             verbose=1,
             n_jobs=-1)

有关为什么这很重要的更多信息，请参阅 this Stack Overflow question/answer