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