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

Question

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

https://github.com/tensorflow/recommenders

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

用户身份	商品编号	评分
2	8	3
5	12	4
6	4	2
...	...	...

我的张量流版本：

张量流==2.4.0

张量流数据集==4.2.0

张量流推荐==0.4.0

Answer 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"])
})