27、基于因子分解机的推荐系统

基于因子分解机的推荐系统

1. 用户电影评分预测

为了实现用户对电影的评分预测，我们定义了一个名为 prediction() 的函数。该函数接收有关用户和物品（这里指电影）的样本输入，通过名称从图中创建 TensorFlow 占位符，然后评估这些张量。需要注意的是，TensorFlow 假定检查点目录已经存在，所以要确保该目录已创建。以下是具体代码：

def prediction(users=predicted_user, items=predicted_item, allow_soft_
placement=allow_soft_placement,\
log_device_placement=log_device_placement, checkpoint_dir=checkpoint_
dir):
    rating_prediction=[]
    checkpoint_prefix = os.path.join(checkpoint_dir, "model")
    graph = tf.Graph()
    with graph.as_default():
        session_conf = tf.ConfigProto(allow_soft_
placement=allow_soft_placement,log_device_placement=log_device_placement)
        with tf.Session(config = session_conf) as sess:
            new_saver = tf.train.import_