簡體   English   中英

帶有參數調整的 Python 包裝器圍繞快速文本訓練

[英]Python wrapper arround fasttext train with parameter tuning

我使用 Fasttext 對有毒評論進行分類(Kaggle 競賽)。 為了訓練我的模型,我運行命令

fasttext supervised -input model_train.train -output model_tune -autotune-validation model_train.valid -autotune-modelsize 100M -autotune-duration 1200

訓練分類模型並進行參數調優,同時保證模型大小在100M以下。 是否有 python 包裝器可以使用-autotune-validation訓練監督模型? 我知道有用於predicttrain方法的 python 包裝器,但找不到使用autotune-validation訓練分類模型的包裝器。 此外,如果最重要的是有一個 sklearn 包裝器,它可以做同樣的事情,那將是非常棒的。

提前致謝

是的,您可以通過向函數添加autotuneValidationFile參數來使用 Python 自動調整它。

參考: https : //fasttext.cc/docs/en/autotune.html

正如解釋在這里,對於fastText自動超參數優化Python包裝的語法如下:

model_tune = fasttext.train_supervised (input='model_train.train', \
  autotuneValidationFile='model_train.valid', autotuneModelSize='100M', \
  autotuneDuration=1200)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM