分类模型预测的类别如何进行识别

Question

我的模型数据中，原始目标变量设置了类别，分别是3，4，5（原来的标准是1-6，但是数据实际分类出来只符合了3,4,5三种类别，其他类别没有数据匹配到）；
而模型预测出来的类别却自动命名为0，1，2。
那我该如何得知他们之间的对应关系呢？
我的代码该如何写？