Python Sklearn:动态设置TfIdfVectorizer的参数

时间:2015-10-27 21:54:15

标签: python scikit-learn

通常我们创建一个TfIdfVectorizer为

TfidfVectorizer(stop_words='english',sublinear_tf=True,use_idf=True)

如果参数在地图中怎么办?有没有办法动态设置TfIdfVectorizer的参数?

1 个答案:

答案 0 :(得分:3)

您可以将参数存储在字典中并从那里设置:

params = {'stop_words':'english','sublinear_tf':True,'use_idf'=True}
TfidfVectorizer(stop_words=params['stop_words'],sublinear_tf=params['sublinear_tf'],use_idf=params['use_idf']).

然而,Python也有一个特殊的语法,允许你在使用dict时传递参数,这将获得与上面相同的结果:

TfidfVectorizer(**params)