如何在Python脚本中更改fasttext API的参数

我们有fasttext命令可以在命令提示符中运行

我已经克隆了GitHub仓库,例如,为了在命令行中更改监督学习网络的参数,我使用了以下命令:

 ./fasttext supervised -input FT_Race_data.txt -output race_model  -lr 0.4 -epoch 30 -loss hs

我更改了学习率(lr)、迭代次数(epoch)和损失函数(loss)。我可以训练模型并获取所需的输出。

在编写Python脚本时,我安装了fasttext库,并尝试了如下代码:

classifier = fasttext.supervised('FT_Race_data.txt','race_model') 

模型被训练了,但结果不理想。在这种情况下,我没有定义任何参数。所以我尝试了如下代码:

classifier = fasttext.supervised('FT_Race_data.txt','race_model', 0.4, 30, 'hs')

程序运行没有错误,但没有给出任何结果。因此我尝试了如下代码:

classifier = fasttext.supervised(input = 'FT_Race_data.txt',output ='race_model', lr = 0.4,epoch= 30,loss = 'hs')

它报错说fasttext只接受两个参数。

如何像在命令提示符中那样在Python脚本中更改参数以微调监督学习?


回答:

为了以后参考,从这里的讨论来看,似乎pip install fasttext并没有安装仓库中可用的所有功能。

因此,在https://pypi.python.org/pypi/fasttext包含最新功能之前,为了使用能够训练模型和设置参数的Python绑定,请按照这里概述的安装程序进行安装。

git clone https://github.com/facebookresearch/fastText.gitcd fastTextpip install .

然后使用train_supervised函数,它返回一个模型对象,可以像以下仓库中的示例那样设置不同的参数。

fastText.train_supervised(input, lr=0.1, dim=100, ws=5, epoch=5, minCount=1, minCountLabel=0, minn=0, maxn=0, neg=5, wordNgrams=1, loss='softmax', bucket=2000000, thread=12, lrUpdateRate=100, t=0.0001, label='__label__', verbose=2, pretrainedVectors='')

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注