我正在使用AWS Java SDK来生成实时预测。为了获取预测输出,我需要知道所使用的机器学习模型类型,是binary
(二元)、regression
(回归)还是multiclass
(多类)。我只有模型ID,可以用它来定位模型。是否有API或其他方法可以在我的应用程序中了解模型类型?
我已经查看了文档,但没有找到符合我需求的信息。
回答:
您可以通过调用GetMLModel API来获取模型类型。
AmazonMachineLearningClient client = ...;GetMLModelRequest request = new GetMLModelRequest().withMLModelId("...");client.getMLModel(request).getMLModelType();