CoreML MLRecommender 问题

Question

我想创建 CoreML Recommender 模型，我的训练数据如下所示：

userID (UUID)  | itemID (UUID) | rating (Int)

用户可以对 0 到 x 的项目数量进行评分，直到训练数据中的每个用户对相同数量的项目（例如 10 个）进行评分，一切正常。

但是当我尝试在哪里创建训练模型时：

• user1 对两个项目进行评分，
• user2 对五个项目进行评分，
• user3 评价一项

我开始遇到这个错误：

Playground execution terminated: An error was thrown and was not caught:
▿ Item IDs in the recommender model must be numbered 0, 1, ..., num_items - 1.
  ▿ generic : 1 element
    - reason : "Item IDs in the recommender model must be numbered 0, 1, ..., num_items - 1."

我该如何解决这个问题？

完整的游乐场代码：

let trainingData = try MLDataTable(contentsOf: URL(fileURLWithPath: "/.../test.csv"))
let model = try MLRecommender (trainingData: trainingData, userColumn: "userID", itemColumn: "itemID", ratingColumn: "rating")
let recs = try model.recommendations(fromUsers: ["050510A3-0C96-4F57-8A45-433422680464"])

Answer 1

我已经设法通过使用大于 0 的标准化和虚拟分数来解决此问题。更多信息：https://pawel.madej.com/post/ml-recommender-in-practice