其他训练参数怎么设置

mayabot / fastText4j

Implementing Facebook's FastText with java

159 stars 23 forks source link

其他训练参数怎么设置 #5

Closed Zhi-lin closed 4 years ago

Zhi-lin commented 5 years ago

请问，FastText.train()除了可以设置文件，模型训练类型之前。其他的参数如何设置，比如学习率，epoch n-gram_window等等。请指教

jimichan commented 5 years ago

        File trFile = new File("data/train.txt");

        TrainArgs trainArgs = new TrainArgs();
        trainArgs.setDim(100);
        trainArgs.setEpoch(100);

        FastText result = FastText.train(trFile,ModelName.cbow,trainArgs);

        result.saveModel("data/out/model.bin");

其中TrainArgs你可以设置很多参数，和原版保持一致。

Zhi-lin commented 5 years ago

谢谢。trainArgs以什么形式传入进去，map吗。对java的多参数机制不是很熟悉

Zhi-lin commented 5 years ago

已实验出来。TrainArgs是个对象，谢谢

jimichan commented 5 years ago

不客气，你们是用FastText做文本分类？

garfieldkai commented 5 years ago

@jimichan 能给介绍一下TrainArgs中各个参数的含义么？谢谢

doudouaili commented 5 years ago

@jimichan 能给介绍一下TrainArgs中各个参数的含义么？谢谢

首页有介绍