word2vec java_Word2vec java实战

作者：AllinToyou | 2024-04-04 05:17:01

踩

java dlp4j word2vec

前言

在学习了word2vec的牛逼后，开始进入实战，解决问题了。

实战

添加依赖

com.medallia.word2vec

word2vecjava_2.11

1.0-ALLENAI-4

训练模型

由于语料比较小，各项参数，都调小了。

@Service

@Slf4j

public class Word2vecService {

public Word2VecModel train() {

try {

List data = List.of("anarchism originated as a term of abuse first used against early working class radicals including the diggers of the english anarchism originated as a term of abuse first");

List list = Lists.transform(data, var11 -> Arrays.asList(var11.split(" ")));

Word2VecModel word2VecModel = Word2VecModel.trainer().setMinVocabFrequency(1).useNumThreads(4).setWindowSize(1).type(NeuralNetworkType.CBOW).setLayerSize(12).useNegativeSamples(5).setDownSamplingRate(1.0E-4D).setNumIterations(5).setListener((var1, var2)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/357372