Searched refs:vocab_weights (Results 1 – 1 of 1) sorted by relevance
251 std::vector<float> vocab_weights; in NegTrainOp() local252 vocab_weights.reserve(vocab_count.size()); in NegTrainOp()255 vocab_weights.push_back(r); in NegTrainOp()257 sampler_ = new random::DistributionSampler(vocab_weights); in NegTrainOp()