19d4eaaf5e1883776b341a4f704bb6a277e315aa,official/recommendation/data_preprocessing.py,,_train_eval_map_fn,#Any#,177

Before Change

    test_blocks.append((
        block_user[0] * np.ones((rconst.NUM_EVAL_NEGATIVES + 1,),
                                dtype=np.int32),
        np.array([block_items[-1]] + test_negatives, dtype=np.uint16)
    ))
    test_positives.append((block_user[0], block_items[-1]))

  train_users = np.concatenate([i[0] for i in train_blocks])

After Change

  with tf.gfile.Open(train_shard_fpath, "wb") as f:
    pickle.dump({
        rconst.TRAIN_KEY: {
            movielens.USER_COLUMN: train_users,
            movielens.ITEM_COLUMN: train_items,
        },

Instances