如何解决如何在 TensorFlow Recommenders 库中使用自定义 .csv 数据集？
我是 tensorflow 的新手。我想使用 TensorFlow Recommenders 库和以下提供的简单代码在我的数据集上训练推荐模型：
https://github.com/tensorflow/recommenders
我想知道如何使用（加载并提供给模型）我的自定义 .csv 文件，而不是加载内置的 Movielens 数据集？
| user_id | item_id | 评分 |
|---|---|---|
| 2 | 8 | 3 |
| 5 | 12 | 4 |
| 6 | 4 | 2 |
| ... | ... | ... |
我的 Tensorflow 版本：
张量流==2.4.0
tensorflow-datasets==4.2.0
tensorflow-recommenders==0.4.0
解决方法
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"])
})
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。