使用pi中的“dataframemapper”类时出错

2024-04-20 05:15:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试保存一个DataFramMapper对象以用于模型的新数据。

 mapper = DataFrameMapper([
        (['price', 'Argentina', 'Canada', 'Australia', 'barcat_numeric'], None),
        ('TTL',CountVectorizer( ngram_range=(1, 2))),
        ('BARCAT', CountVectorizer( ngram_range=(1, 2)))
    ])
    with open('company_dill.pkl', 'wb') as f:
        dill.dump(mapper, f)

当我读入数据时:

^{pr2}$

输出为:
但当我尝试使用它时,我得到:

^{3}$

当我不保存到pickle时,它可以正常工作,同时使用pickle和{}


Tags: 数据对象模型rangepricepicklemapperdill
1条回答
网友
1楼 · 发布于 2024-04-20 05:15:42

好吧,我发现了我的错误: 在拟合数据之前,我保存了DataFrameMapper对象,这是正确的方法:

 mapper = DataFrameMapper([
        (['price', 'Argentina', 'Canada', 'Australia', 'barcat_numeric'], None),
        ('TTL',CountVectorizer( ngram_range=(1, 2))),
        ('BARCAT', CountVectorizer( ngram_range=(1, 2)))
    ])
    X=mapper.fit_transform(data_frame)
    pickle.dump(mapper, open( "mapper.pkl", "wb") )

相关问题 更多 >