Java Code Examples for org.dmg.pmml.OpType#CONTINUOUS

static
public OpType getOpType(Output output){
	org.tensorflow.DataType dataType = output.dataType();

	switch(dataType){
		case FLOAT:
		case DOUBLE:
		case INT32:
		case INT64:
			return OpType.CONTINUOUS;
		case STRING:
		case BOOL:
			return OpType.CATEGORICAL;
		default:
			throw new IllegalArgumentException();
	}
}
 

static
public OpType getOpType(DataType dataType){

	switch(dataType){
		case STRING:
			return OpType.CATEGORICAL;
		case INTEGER:
		case FLOAT:
		case DOUBLE:
			return OpType.CONTINUOUS;
		case BOOLEAN:
			return OpType.CATEGORICAL;
		case DATE:
		case DATE_TIME:
			return OpType.ORDINAL;
		default:
			throw new IllegalArgumentException();
	}
}
 

static
public FieldValue create(DataType dataType, Object value){

	if(value instanceof Collection){
		return new CollectionValue(dataType, OpType.CONTINUOUS, (Collection<?>)value);
	}

	switch(dataType){
		case INTEGER:
			return new ContinuousInteger(value);
		case FLOAT:
			return new ContinuousFloat(value);
		case DOUBLE:
			return new ContinuousDouble(value);
		default:
			return new ContinuousValue(dataType, value);
	}
}
 

public static DataDictionary buildDataDictionary(
    InputSchema schema,
    CategoricalValueEncodings categoricalValueEncodings) {
  List<String> featureNames = schema.getFeatureNames();

  List<DataField> dataFields = new ArrayList<>();
  for (int featureIndex = 0; featureIndex < featureNames.size(); featureIndex++) {
    String featureName = featureNames.get(featureIndex);
    OpType opType;
    DataType dataType;
    if (schema.isNumeric(featureName)) {
      opType = OpType.CONTINUOUS;
      dataType = DataType.DOUBLE;
    } else if (schema.isCategorical(featureName)) {
      opType = OpType.CATEGORICAL;
      dataType = DataType.STRING;
    } else {
      // Don't know
      opType = null;
      dataType = null;
    }
    DataField field = new DataField(FieldName.create(featureName), opType, dataType);
    if (schema.isCategorical(featureName)) {
      Objects.requireNonNull(categoricalValueEncodings);
      categoricalValueEncodings.getEncodingValueMap(featureIndex).entrySet().stream().
          sorted(Comparator.comparing(Map.Entry::getKey)).
          map(Map.Entry::getValue).
          forEach(value -> field.addValues(new Value(value)));
    }
    dataFields.add(field);
  }

  return new DataDictionary(dataFields).setNumberOfFields(dataFields.size());
}
 

static
public OpType getOpType(DataType dataType){

	switch(dataType){
		case STRING:
			return OpType.CATEGORICAL;
		case INTEGER:
		case FLOAT:
		case DOUBLE:
			return OpType.CONTINUOUS;
		case BOOLEAN:
			return OpType.CATEGORICAL;
		case DATE:
		case TIME:
		case DATE_TIME:
			return OpType.ORDINAL;
		case DATE_DAYS_SINCE_0:
		case DATE_DAYS_SINCE_1960:
		case DATE_DAYS_SINCE_1970:
		case DATE_DAYS_SINCE_1980:
		case TIME_SECONDS:
		case DATE_TIME_SECONDS_SINCE_0:
		case DATE_TIME_SECONDS_SINCE_1960:
		case DATE_TIME_SECONDS_SINCE_1970:
		case DATE_TIME_SECONDS_SINCE_1980:
			return OpType.CONTINUOUS;
		default:
			throw new IllegalArgumentException();
	}
}
 

@Override
public OpType getOpType(){
	return OpType.CONTINUOUS;
}
 

@Override
public OpType getOpType(){
	return OpType.CONTINUOUS;
}
 

@Override
public OpType getOpType(){
	return OpType.CONTINUOUS;
}
 

@Override
public OpType getOpType(){
	return OpType.CONTINUOUS;
}
 

@Override
public OpType getOpType(){
	return OpType.CONTINUOUS;
}
 

@Test
public void getName(){
	FieldName name = FieldName.create("y");

	DataField dataField = new DataField(name, OpType.CONTINUOUS, DataType.DOUBLE);

	TargetField targetField = new TargetField(dataField, null, null);

	assertEquals(name, targetField.getName());

	targetField.setName(FieldName.create("label"));

	assertNotEquals(name, targetField.getName());

	targetField.setName(null);

	assertEquals(name, targetField.getName());
}
 

@Test
public void getOpType(){
	FieldName name = FieldName.create("y");

	DataField dataField = new DataField(name, OpType.CONTINUOUS, DataType.DOUBLE);

	MiningField miningField = new MiningField(name)
		.setOpType(OpType.CATEGORICAL);

	Target target = new Target()
		.setField(name)
		.setOpType(OpType.CONTINUOUS);

	TargetField targetField = new TargetField(dataField, null, null);

	assertEquals(OpType.CONTINUOUS, targetField.getOpType());

	targetField = new TargetField(dataField, miningField, null);

	assertEquals(OpType.CATEGORICAL, targetField.getOpType());

	targetField = new TargetField(dataField, miningField, target);

	assertEquals(OpType.CONTINUOUS, targetField.getOpType());
}
 

@Test
public void isDefault(){
	FieldName name = FieldName.create("x");

	DataField dataField = new DataField(name, OpType.CONTINUOUS, DataType.DOUBLE);

	MiningField miningField = new MiningField(name);

	assertTrue(InputFieldUtil.isDefault(dataField, miningField));

	miningField.setOpType(OpType.CATEGORICAL);

	assertFalse(InputFieldUtil.isDefault(dataField, miningField));
}