Tensorflow Estimator API embedding_column计算邻域

I am trying to implement a similar model to word2vec in tensorflow using the latest estimator API. The problem I am running into is when I try to evaluate the model. The model itself should be the basis for a recommender system. Now the evaluation metric I would like to use is the Hitrate metric, as follows:I am trying to implement a similar model to wor