private static void checkDataField(DataField field, String name, Boolean categorical) { assertEquals(name, field.getName().getValue()); if (categorical == null) { assertNull(field.getOpType()); assertNull(field.getDataType()); } else if (categorical) { assertEquals(OpType.CATEGORICAL, field.getOpType()); assertEquals(DataType.STRING, field.getDataType()); } else { assertEquals(OpType.CONTINUOUS, field.getOpType()); assertEquals(DataType.DOUBLE, field.getDataType()); } }
protected static void checkDataDictionary(InputSchema schema, DataDictionary dataDictionary) { assertNotNull(dataDictionary); assertEquals("Wrong number of features", schema.getNumFeatures(), dataDictionary.getNumberOfFields().intValue()); List<DataField> dataFields = dataDictionary.getDataFields(); assertEquals(schema.getNumFeatures(), dataFields.size()); for (DataField dataField : dataFields) { String featureName = dataField.getName().getValue(); if (schema.isNumeric(featureName)) { assertEquals("Wrong op type for feature " + featureName, OpType.CONTINUOUS, dataField.getOpType()); assertEquals("Wrong data type for feature " + featureName, DataType.DOUBLE, dataField.getDataType()); } else if (schema.isCategorical(featureName)) { assertEquals("Wrong op type for feature " + featureName, OpType.CATEGORICAL, dataField.getOpType()); assertEquals("Wrong data type for feature " + featureName, DataType.STRING, dataField.getDataType()); } else { assertNull(dataField.getOpType()); assertNull(dataField.getDataType()); } } }
public void setLabel(DataField dataField){ Label label; OpType opType = dataField.getOpType(); switch(opType){ case CATEGORICAL: label = new CategoricalLabel(dataField); break; case CONTINUOUS: label = new ContinuousLabel(dataField); break; default: throw new IllegalArgumentException(); } setLabel(label); }
public void addFeature(DataField dataField){ OpType opType = dataField.getOpType(); switch(opType){ case CONTINUOUS: addFeature(new ContinuousFeature(this, dataField)); break; case CATEGORICAL: addFeature(new CategoricalFeature(this, dataField)); break; default: throw new IllegalArgumentException(); } }
public void setLabel(DataField dataField){ OpType opType = dataField.getOpType(); switch(opType){ case CONTINUOUS: setLabel(new ContinuousLabel(dataField)); break; case CATEGORICAL: setLabel(new CategoricalLabel(dataField)); break; default: throw new IllegalArgumentException(); } }
OpType opType = dataField.getOpType(); switch(opType){ case CATEGORICAL:
OpType opType = dataField.getOpType(); switch(opType){ case CATEGORICAL:
HasParsedValueMapping<?> hasParsedValueMapping = (HasParsedValueMapping<?>)dataField; OpType opType = dataField.getOpType(); if(opType == null){ throw new MissingAttributeException(dataField, PMMLAttributes.DATAFIELD_OPTYPE);