我们有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='')