Java ImportFilter类代码示例

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
@Override
public void init(ProcessContext processContext) throws Exception {
    //load model and marshal it into pmml version  > 4.0
    log.info("Loading .pmml model");
    try (InputStream is = url.openStream()) {
        Source transformedSource = ImportFilter.apply(new InputSource(is));
        pmml = JAXBUtil.unmarshalPMML(transformedSource);
    } catch (SAXException ex) {
        log.error("Could not load model from file provided at" + url);
    }
    //build a modelevaluator from the loaded pmml file
    ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();
    modelEvaluator = modelEvaluatorFactory.newModelEvaluator(pmml);


    log.info("Loaded model requires the following fields: " + modelEvaluator.getActiveFields().toString());

    log.info("Loaded model has targets: " + modelEvaluator.getTargetFields().toString());
    if (modelEvaluator.getTargetFields().size() > 1) {
        log.error("Only models with one target variable are supported for now");
    }

    targetName = modelEvaluator.getTargetFieldName();
    activeFields = modelEvaluator.getActiveFields();
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
private void transform(File pmmlFile, File serFile, List<Class<? extends Visitor>> visitorClazzes) throws Exception {
	PMML pmml;

	try(InputStream is = new FileInputStream(pmmlFile)){
		Source source = ImportFilter.apply(new InputSource(is));

		pmml = JAXBUtil.unmarshalPMML(source);
	}

	for(Class<? extends Visitor> visitorClazz : visitorClazzes){
		Visitor visitor = visitorClazz.newInstance();

		visitor.applyTo(pmml);
	}

	try(OutputStream os = new FileOutputStream(serFile)){
		SerializationUtil.serializePMML(pmml, os);
	}
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
static
public Evaluator createEvaluator(InputStream is) throws SAXException, JAXBException {
	Source source = ImportFilter.apply(new InputSource(is));

	PMML pmml = JAXBUtil.unmarshalPMML(source);

	// If the SAX Locator information is available, then transform it to java.io.Serializable representation
	LocatorTransformer locatorTransformer = new LocatorTransformer();
	locatorTransformer.applyTo(pmml);

	ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();

	ModelEvaluator<?> modelEvaluator = modelEvaluatorFactory.newModelManager(pmml);
	modelEvaluator.verify();

	return modelEvaluator;
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
/**
 * @param pmmlString PMML model encoded as an XML doc in a string
 * @return {@link PMML} object representing the model
 * @throws IOException if XML can't be unserialized
 */
public static PMML fromString(String pmmlString) throws IOException {
  // Emulate PMMLUtil.unmarshal here, but need to accept a Reader
  InputSource source = new InputSource(new StringReader(pmmlString));
  try {
    return JAXBUtil.unmarshalPMML(ImportFilter.apply(source));
  } catch (JAXBException | SAXException e) {
    throw new IOException(e);
  }
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
@PostConstruct
public void setUp() throws IOException, SAXException, JAXBException {
	try (InputStream is = properties.getModelLocation().getInputStream()) {
		Source transformedSource = ImportFilter.apply(new InputSource(is));
		pmml = JAXBUtil.unmarshalPMML(transformedSource);
		Assert.state(!pmml.getModels().isEmpty(),
				"The provided PMML file at " + properties.getModelLocation() + " does not contain any model");
	}
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
static
private PMML unmarshal(File file) throws Exception {

	try(InputStream is = new FileInputStream(file)){
		Source source = ImportFilter.apply(new InputSource(is));

		return JAXBUtil.unmarshalPMML(source);
	}
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
private static Evaluator createEvaluator(String filePath) throws SAXException, JAXBException, IOException{
	InputStream is = new FileInputStream(new File(filePath));
	PMML pmml;
	try {
		Source source = ImportFilter.apply(new InputSource(is));
		pmml = JAXBUtil.unmarshalPMML(source);
	} finally {
		is.close();
	}
	PMMLManager pmmlManager = new PMMLManager(pmml);
	return (Evaluator)pmmlManager.getModelManager(ModelEvaluatorFactory.getInstance());		
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
static
public Evaluator createEvaluator(InputStream is) throws SAXException, JAXBException {
	Source source = ImportFilter.apply(new InputSource(is));

	PMML pmml = JAXBUtil.unmarshalPMML(source);

	// If the SAX Locator information is available, then transform it to java.io.Serializable representation
	LocatorTransformer locatorTransformer = new LocatorTransformer();
	locatorTransformer.applyTo(pmml);

	ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();

	Evaluator evaluator = modelEvaluatorFactory.newModelEvaluator(pmml);

	// Perform self-testing
	evaluator.verify();

	return evaluator;
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
static
public Evaluator createEvaluator(InputStream is) throws SAXException, JAXBException {
	Source source = ImportFilter.apply(new InputSource(is));

	PMML pmml = JAXBUtil.unmarshalPMML(source);

	// If the SAX Locator information is available, then transform it to java.io.Serializable representation
	LocatorTransformer locatorTransformer = new LocatorTransformer();
	pmml.accept(locatorTransformer);

	ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();

	ModelEvaluator<?> modelEvaluator = modelEvaluatorFactory.newModelManager(pmml);

	return modelEvaluator;
}
 

import org.jpmml.model.ImportFilter; //导入依赖的package包/类
static
private ModelEvaluator<?> unmarshal(InputStream is, boolean validate) throws IOException, SAXException, JAXBException {
	XMLReader reader = XMLReaderFactory.createXMLReader();
	reader.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);

	ImportFilter filter = new ImportFilter(reader);

	Source source = new SAXSource(filter, new InputSource(is));

	Unmarshaller unmarshaller = JAXBUtil.createUnmarshaller();
	unmarshaller.setEventHandler(new SimpleValidationEventHandler());

	if(validate){
		Schema schema = JAXBUtil.getSchema();

		unmarshaller.setSchema(schema);
	}

	PMML pmml = (PMML)unmarshaller.unmarshal(source);

	ModelEvaluatorFactory modelEvaluatorFactory = ModelEvaluatorFactory.newInstance();

	return modelEvaluatorFactory.newModelEvaluator(pmml);
}