如何在tensorflow js中构建项目推荐模型

Question

我正在为英雄联盟构建一个基于机器学习的项目推荐系统。 我的目标是将模型传递给游戏状态（带有项目的玩家列表），返回一组推荐的购买项目。

我的训练数据是球员名单以及他们在什么时间买的项目。将在游戏的每一分钟传递此快照。 查询

[
    [teamId, positionId, championId, level, gold, kills, deaths, assists, itemId1, itemId2, itemId3, itemId4, itemId5, itemId6],
    [teamId, positionId, championId, level, gold, kills, deaths, assists, itemId1, itemId2, itemId3, itemId4, itemId5, itemId6],
    ...
]

答案是推荐项目列表

[itemId1, itemId2, itemId3, itemId4, itemId5, itemId6]

玩家将始终按此顺序传递

[
    player
    ally 1
    ally 2
    ally 3
    ally 4
    enemy 1
    enemy 2
    enemy 3
    enemy 4
    enemy 5
]

在训练期间，我将始终首先列出获胜队伍并使用第一个队员最终构建作为输出。

当我开始写这个问题时，我不知道从哪里开始。写这个确实有助于提出一个总体计划。所以...

1. 这是接近我想做的事情的好方法吗？
2. 我有什么要改变的吗？

对于那些不了解https://js.tensorflow.org/

的人