如何在 TensorFlow Recommenders 库中使用自定义 .csv 数据集？答案

【问题标题】：How to use a custom .csv dataset in TensorFlow Recommenders library?如何在 TensorFlow Recommenders 库中使用自定义 .csv 数据集？
【发布时间】：2021-06-26 14:27:31
【问题描述】：

我是张量流的新手。我想使用 TensorFlow Recommenders 库和提供的简单代码在我的数据集上训练推荐模型：

https://github.com/tensorflow/recommenders

我想知道如何使用（加载并提供给模型）以下格式的自定义 .csv 文件，而不是加载内置的 Movielens 数据集？

user_id	item_id	rating
2	8	3
5	12	4
6	4	2
...	...	...

我的 TensorFlow 版本：

张量流==2.4.0

tensorflow-datasets==4.2.0

tensorflow-recommenders==0.4.0

【问题讨论】：

标签： python tensorflow tensorflow-datasets recommendation-engine recommender-systems

【解决方案1】：

import pandas as pd

DATA_URL = "D:/ratings.csv"
df = pd.read_csv(DATA_URL)
ratings = tf.data.Dataset.from_tensor_slices(dict(df)).map(lambda x: {
    "user_id": str(x["user_id"]),
    "item_id": str(x["item_id"]),
    "rating": float(x["rating"])
})

【讨论】：

时间戳怎么样？