in src/main/java/com/amazonaws/sagemaker/helper/DataConversionHelper.java [208:254]
protected DataType convertInputToMLeapInputType(final String type, final String structure) {
BasicType basicType;
switch (type) {
case BasicDataType.INTEGER:
basicType = support.createInt();
break;
case BasicDataType.LONG:
basicType = support.createLong();
break;
case BasicDataType.FLOAT:
basicType = support.createFloat();
break;
case BasicDataType.DOUBLE:
basicType = support.createDouble();
break;
case BasicDataType.BOOLEAN:
basicType = support.createBoolean();
break;
case BasicDataType.BYTE:
basicType = support.createByte();
break;
case BasicDataType.SHORT:
basicType = support.createShort();
break;
case BasicDataType.STRING:
basicType = support.createString();
break;
default:
basicType = null;
}
if (basicType == null) {
throw new IllegalArgumentException("Data type passed in the request is wrong for one or more columns");
}
if (StringUtils.isNotBlank(structure)) {
switch (structure) {
case DataStructureType.VECTOR:
return new TensorType(basicType, true);
case DataStructureType.ARRAY:
return new ListType(basicType, true);
case DataStructureType.BASIC:
return new ScalarType(basicType, true);
}
}
// if structure field is not passed, it is by default basic
return new ScalarType(basicType, true);
}