我有两个模型可以嵌入两本书

Ml_model_a => book1_embedding [ 1, 200 ]

Ml_model_b => book2_embedding [ 1, 200 ]

我正在构建第三个模型，它将采用这两种不同的嵌入来告诉我该选择哪本书。

现在我的最后一层是 0,1 之间的分类（选择哪本书）。如何以最好的方式学习这些嵌入以更好地分类？

我已经尝试过的：

如果我对这些嵌入进行平均然后发送到一个模型，那么嵌入会丢失很多信息，所以我使用连接方法。

但是分类不好，有没有其他模型，一种我可以用来增强学习书籍嵌入和预测要拿哪本书的能力的技术？