Java JAXBUtil类代码示例

import org.jpmml.model.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的package包/类
@Test
public void testUncompressed() throws Exception {
    ModelConfig modelConfig = setupConfig();
    RManager rManager = setupManager();

    UUID uuid  = rManager.trainAndAdd(modelConfig, RIntegrationTest.getTrainingInstances());

    File targetFile = Files.createTempFile("targetPMML", ".xml").toFile();

    // Save the model as PMML and load it.
    rManager.saveAsPMML(uuid, targetFile.getAbsolutePath(), false);

    try (FileInputStream fis = new FileInputStream(targetFile)) {
        JAXBUtil.unmarshalPMML(new StreamSource(fis));
    }

    targetFile.delete();
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
@Test
public void testCompressed() throws Exception {
    ModelConfig modelConfig = setupConfig();
    RManager rManager = setupManager();

    UUID uuid  = rManager.trainAndAdd(modelConfig, RIntegrationTest.getTrainingInstances());

    File targetFile = Files.createTempFile("targetPMML", ".xml").toFile();

    // Save the model as PMML and load it.
    rManager.saveAsPMML(uuid, targetFile.getAbsolutePath(), true);

    try (FileInputStream fis = new FileInputStream(targetFile);
         GZIPInputStream gis = new GZIPInputStream(fis)) {

        JAXBUtil.unmarshalPMML(new StreamSource(gis));
    }

    targetFile.delete();
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的package包/类
/**
 * @param pmml model
 * @return model serialized as an XML document as a string
 */
public static String toString(PMML pmml) {
  try (StringWriter out = new StringWriter()) {
    // v JAXBUtil.marshalPMML but need to set compact, non-pretty output
    Marshaller marshaller = JAXBUtil.createMarshaller();
    marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.FALSE);
    marshaller.marshal(pmml, new StreamResult(out));
    return out.toString();
  } catch (JAXBException | IOException e) {
    // IOException should not be possible; JAXBException would only happen with XML
    // config problems.
    throw new IllegalStateException(e);
  }
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的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.JAXBUtil; //导入依赖的package包/类
public static String convertPMMLToString(PMML pmml) throws JAXBException, UnsupportedEncodingException {
    ByteArrayOutputStream baor = new ByteArrayOutputStream();
    StreamResult streamResult = new StreamResult();
    streamResult.setOutputStream(baor);
    JAXBUtil.marshal(pmml, streamResult);
    return baor.toString(Charset.defaultCharset().toString());
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
/**
 * Test model library with sample training and test data
 */
@TestWith({"testMetaDataAll"})
public void testModelLibrary(TestMetaData testMetaData) throws Exception {
  Driver driver = new com.cloudera.framework.example.three.Driver(dfsServer.getConf());
  assertEquals(SUCCESS, driver.runner(dfsServer.getPath(DATASET_DIR).toString()));
  assertTrue(dfsServer.getFileSystem().exists(dfsServer.getPath(DATASET_DIR + "/" + ModelDir() + "/" + ModelFile())));
  assertTrue(Double.parseDouble(((PMML) JAXBUtil.unmarshal(new StreamSource(new StringReader((String) driver.getResults().get(0))))).
    getHeader().getExtensions().get(0).getContent().get(0).toString()) > ModelPmml.MinimumAccuracy());
  assertTrue(Double.parseDouble(((PMML) JAXBUtil.unmarshal(new StreamSource(dfsServer.getFileSystem().open(
    dfsServer.getPath(DATASET_DIR + "/" + ModelDir() + "/" + ModelFile()))))).
    getHeader().getExtensions().get(0).getContent().get(0).toString()) > ModelPmml.MinimumAccuracy());
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
/**
 * Test model script with sample training and test data
 */
@Test
public void testModelScript() throws Exception {
  assertEquals(0, scalaServer.execute(new File(DIR_SOURCE_SCRIPT, SCRIPT_SCALA)));
  assertTrue(dfsServer.getFileSystem().exists(dfsServer.getPath(DATASET_DIR + "/" + ModelDir() + "/" + ModelFile())));
  assertTrue(Double.parseDouble(((PMML) JAXBUtil.unmarshal(new StreamSource(dfsServer.getFileSystem().open(
    dfsServer.getPath(DATASET_DIR + "/" + ModelDir() + "/" + ModelFile()))))).
    getHeader().getExtensions().get(0).getContent().get(0).toString()) > ModelPmml.MinimumAccuracy());
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
@Override
public FastRandomForest consume(String pmmlString) throws PMMLConversionException {
    PMML pmml = null;
    try (ByteArrayInputStream bais = new ByteArrayInputStream(pmmlString.getBytes())){
        pmml = JAXBUtil.unmarshalPMML(new StreamSource(bais));
    } catch (Exception e) {
        throw new PMMLConversionException("Failed to unmarshal PMML from string. Make sure it is a valid PMML.", e);
    }
    return consume(pmml);
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
@Override
public void produce(FastRandomForest randomForestClassifier, File targetFile) throws PMMLConversionException {
    PMML pmml = produce(randomForestClassifier);
    try (FileOutputStream fis = new FileOutputStream(targetFile)){
        JAXBUtil.marshalPMML(pmml, new StreamResult(fis));
    } catch (Exception e) {
        throw new PMMLConversionException("Failed to marshal the PMML to the given file.", e);
    }
}
 

import org.jpmml.model.JAXBUtil; //导入依赖的package包/类
@Override
public RandomForest consume(String pmmlString) throws PMMLConversionException {
    PMML pmml = null;
    try (ByteArrayInputStream bais = new ByteArrayInputStream(pmmlString.getBytes())){
        pmml = JAXBUtil.unmarshalPMML(new StreamSource(bais));
    } catch (Exception e) {
        throw new PMMLConversionException("Failed to unmarshal PMML from string. Make sure it is a valid PMML.", e);
    }
    return consume(pmml);
}