in src/main/java/com/amazonaws/sagemaker/helper/DataConversionHelper.java [121:141]
public Object convertMLeapBasicTypeToJavaType(final ArrayRow predictionRow, final String type) {
switch (type) {
case BasicDataType.INTEGER:
return predictionRow.getInt(0);
case BasicDataType.LONG:
return predictionRow.getLong(0);
case BasicDataType.FLOAT:
case BasicDataType.DOUBLE:
return predictionRow.getDouble(0);
case BasicDataType.BOOLEAN:
return predictionRow.getBool(0);
case BasicDataType.BYTE:
return predictionRow.getByte(0);
case BasicDataType.SHORT:
return predictionRow.getShort(0);
case BasicDataType.STRING:
return predictionRow.getString(0);
default:
throw new IllegalArgumentException("Given type is not supported");
}
}