使用后，get_preds我得到一个张量列表。就像是：
[
[0.9, 0.1],
[0.85, 0.15],
[0.92, 0.08],
...
]
我的班级是[0,1]。我应该如何将这些张量转换为相应的（最有可能的）类？请在下面查看我目前的方法
probs = learn.get_preds(ds_type=DatasetType.Test)[0]
def probs2class(item):
return max(range(len(item)), key=item.__getitem__)
print(map(probs2class, probs))
我疯狂地搜索文档，但可能是错误的术语？从概率到类预测的一般方法是什么？