如何在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

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

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