Search in sources :

Example 6 with ModelReader

use of org.apache.ignite.ml.inference.reader.ModelReader in project ignite by apache.

the class XGBoostModelParserExample method main.

/**
 * Run example.
 */
public static void main(String... args) throws ExecutionException, InterruptedException, FileNotFoundException {
    try (Ignite ignite = Ignition.start("examples/config/example-ignite.xml")) {
        File mdlRsrc = IgniteUtils.resolveIgnitePath(TEST_MODEL_RES);
        if (mdlRsrc == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_MODEL_RES + "]");
        ModelReader reader = new FileSystemModelReader(mdlRsrc.getPath());
        AsyncModelBuilder mdlBuilder = new IgniteDistributedModelBuilder(ignite, 4, 4);
        File testData = IgniteUtils.resolveIgnitePath(TEST_DATA_RES);
        if (testData == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_DATA_RES + "]");
        File testExpRes = IgniteUtils.resolveIgnitePath(TEST_ER_RES);
        if (testExpRes == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_ER_RES + "]");
        try (Model<NamedVector, Future<Double>> mdl = mdlBuilder.build(reader, parser);
            Scanner testDataScanner = new Scanner(testData);
            Scanner testExpResultsScanner = new Scanner(testExpRes)) {
            while (testDataScanner.hasNextLine()) {
                String testDataStr = testDataScanner.nextLine();
                String testExpResultsStr = testExpResultsScanner.nextLine();
                HashMap<String, Double> testObj = new HashMap<>();
                for (String keyValueString : testDataStr.split(" ")) {
                    String[] keyVal = keyValueString.split(":");
                    if (keyVal.length == 2)
                        testObj.put("f" + keyVal[0], Double.parseDouble(keyVal[1]));
                }
                double prediction = mdl.predict(VectorUtils.of(testObj)).get();
                double expPrediction = Double.parseDouble(testExpResultsStr);
                System.out.println("Expected: " + expPrediction + ", prediction: " + prediction);
            }
        }
    } finally {
        System.out.flush();
    }
}
Also used : Scanner(java.util.Scanner) HashMap(java.util.HashMap) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) ModelReader(org.apache.ignite.ml.inference.reader.ModelReader) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) NamedVector(org.apache.ignite.ml.math.primitives.vector.NamedVector) AsyncModelBuilder(org.apache.ignite.ml.inference.builder.AsyncModelBuilder) Future(java.util.concurrent.Future) Ignite(org.apache.ignite.Ignite) IgniteDistributedModelBuilder(org.apache.ignite.ml.inference.builder.IgniteDistributedModelBuilder) File(java.io.File)

Example 7 with ModelReader

use of org.apache.ignite.ml.inference.reader.ModelReader in project ignite by apache.

the class H2OMojoModelParserExample method main.

/**
 * Run example.
 */
public static void main(String... args) throws ExecutionException, InterruptedException, FileNotFoundException {
    try (Ignite ignite = Ignition.start("examples/config/example-ignite.xml")) {
        File mdlRsrc = IgniteUtils.resolveIgnitePath(MODEL_RES);
        if (mdlRsrc == null)
            throw new IllegalArgumentException("File not found [resource_path=" + MODEL_RES + "]");
        ModelReader reader = new FileSystemModelReader(mdlRsrc.getPath());
        AsyncModelBuilder mdlBuilder = new IgniteDistributedModelBuilder(ignite, 4, 4);
        File testData = IgniteUtils.resolveIgnitePath(DATA_RES);
        if (testData == null)
            throw new IllegalArgumentException("File not found [resource_path=" + DATA_RES + "]");
        try (Model<NamedVector, Future<Double>> mdl = mdlBuilder.build(reader, parser);
            Scanner testDataScanner = new Scanner(testData)) {
            while (testDataScanner.hasNextLine()) {
                String testDataStr = testDataScanner.nextLine();
                String actual = null;
                HashMap<String, Double> testObj = new HashMap<>();
                for (String keyValueString : testDataStr.split(" ")) {
                    String[] keyVal = keyValueString.split(":");
                    if (keyVal.length == 2)
                        testObj.put("C" + (1 + Integer.parseInt(keyVal[0])), Double.parseDouble(keyVal[1]));
                    else
                        actual = keyValueString;
                }
                double prediction = mdl.predict(VectorUtils.of(testObj)).get();
                System.out.println("Actual: " + actual + ", prediction: " + prediction);
            }
        }
    } finally {
        System.out.flush();
    }
}
Also used : Scanner(java.util.Scanner) HashMap(java.util.HashMap) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) ModelReader(org.apache.ignite.ml.inference.reader.ModelReader) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) NamedVector(org.apache.ignite.ml.math.primitives.vector.NamedVector) AsyncModelBuilder(org.apache.ignite.ml.inference.builder.AsyncModelBuilder) Future(java.util.concurrent.Future) Ignite(org.apache.ignite.Ignite) IgniteDistributedModelBuilder(org.apache.ignite.ml.inference.builder.IgniteDistributedModelBuilder) File(java.io.File)

Example 8 with ModelReader

use of org.apache.ignite.ml.inference.reader.ModelReader in project ignite by apache.

the class CatboostRegressionModelParserExample method main.

/**
 * Run example.
 */
public static void main(String... args) throws ExecutionException, InterruptedException, FileNotFoundException {
    try (Ignite ignite = Ignition.start("examples/config/example-ignite.xml")) {
        File mdlRsrc = IgniteUtils.resolveIgnitePath(TEST_MODEL_RES);
        if (mdlRsrc == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_MODEL_RES + "]");
        ModelReader reader = new FileSystemModelReader(mdlRsrc.getPath());
        AsyncModelBuilder mdlBuilder = new IgniteDistributedModelBuilder(ignite, 4, 4);
        File testData = IgniteUtils.resolveIgnitePath(TEST_DATA_RES);
        if (testData == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_DATA_RES + "]");
        File testExpRes = IgniteUtils.resolveIgnitePath(TEST_ER_RES);
        if (testExpRes == null)
            throw new IllegalArgumentException("File not found [resource_path=" + TEST_ER_RES + "]");
        try (Model<NamedVector, Future<Double>> mdl = mdlBuilder.build(reader, parser);
            Scanner testDataScanner = new Scanner(testData);
            Scanner testExpResultsScanner = new Scanner(testExpRes)) {
            String[] columns = new String[] { "f_0", "f_1", "f_2", "f_3", "f_4", "f_5", "f_6", "f_7", "f_8", "f_9", "f_10", "f_11", "f_12" };
            while (testDataScanner.hasNextLine()) {
                String testDataStr = testDataScanner.nextLine();
                String testExpResultsStr = testExpResultsScanner.nextLine();
                HashMap<String, Double> testObj = new HashMap<>();
                String[] values = testDataStr.split(",");
                for (int i = 0; i < columns.length; i++) {
                    testObj.put(columns[i], Double.valueOf(values[i]));
                }
                double prediction = mdl.predict(VectorUtils.of(testObj)).get();
                double expPrediction = Double.parseDouble(testExpResultsStr);
                System.out.println("Expected: " + expPrediction + ", prediction: " + prediction);
            }
        }
    } finally {
        System.out.flush();
    }
}
Also used : Scanner(java.util.Scanner) HashMap(java.util.HashMap) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) ModelReader(org.apache.ignite.ml.inference.reader.ModelReader) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) NamedVector(org.apache.ignite.ml.math.primitives.vector.NamedVector) AsyncModelBuilder(org.apache.ignite.ml.inference.builder.AsyncModelBuilder) Future(java.util.concurrent.Future) Ignite(org.apache.ignite.Ignite) IgniteDistributedModelBuilder(org.apache.ignite.ml.inference.builder.IgniteDistributedModelBuilder) File(java.io.File)

Example 9 with ModelReader

use of org.apache.ignite.ml.inference.reader.ModelReader in project ignite by apache.

the class CatboostRegressionModelParserTest method testParseAndPredict.

/**
 * End-to-end test for {@code parse()} and {@code predict()} methods.
 */
@Test
public void testParseAndPredict() {
    URL url = CatboostRegressionModelParserTest.class.getClassLoader().getResource(TEST_MODEL_RESOURCE);
    if (url == null)
        throw new IllegalStateException("File not found [resource_name=" + TEST_MODEL_RESOURCE + "]");
    ModelReader reader = new FileSystemModelReader(url.getPath());
    try (CatboostRegressionModel mdl = mdlBuilder.build(reader, parser)) {
        HashMap<String, Double> input = new HashMap<>();
        input.put("f_0", 0.02731d);
        input.put("f_1", 0.0d);
        input.put("f_2", 7.07d);
        input.put("f_3", 0d);
        input.put("f_4", 0.469d);
        input.put("f_5", 6.421d);
        input.put("f_6", 78.9d);
        input.put("f_7", 4.9671d);
        input.put("f_8", 2d);
        input.put("f_9", 242.0d);
        input.put("f_10", 17.8d);
        input.put("f_11", 396.9d);
        input.put("f_12", 9.14d);
        double prediction = mdl.predict(VectorUtils.of(input));
        assertEquals(21.164552741740483, prediction, 1e-5);
    }
}
Also used : CatboostRegressionModel(org.apache.ignite.ml.catboost.CatboostRegressionModel) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) ModelReader(org.apache.ignite.ml.inference.reader.ModelReader) HashMap(java.util.HashMap) FileSystemModelReader(org.apache.ignite.ml.inference.reader.FileSystemModelReader) URL(java.net.URL) Test(org.junit.Test)

Aggregations

ModelReader (org.apache.ignite.ml.inference.reader.ModelReader)9 HashMap (java.util.HashMap)8 FileSystemModelReader (org.apache.ignite.ml.inference.reader.FileSystemModelReader)8 Scanner (java.util.Scanner)5 Future (java.util.concurrent.Future)5 Ignite (org.apache.ignite.Ignite)5 IgniteDistributedModelBuilder (org.apache.ignite.ml.inference.builder.IgniteDistributedModelBuilder)5 File (java.io.File)4 URL (java.net.URL)4 AsyncModelBuilder (org.apache.ignite.ml.inference.builder.AsyncModelBuilder)4 NamedVector (org.apache.ignite.ml.math.primitives.vector.NamedVector)4 Test (org.junit.Test)4 Cache (javax.cache.Cache)1 IgniteCache (org.apache.ignite.IgniteCache)1 SandboxMLCache (org.apache.ignite.examples.ml.util.SandboxMLCache)1 CatboostClassificationModel (org.apache.ignite.ml.catboost.CatboostClassificationModel)1 CatboostRegressionModel (org.apache.ignite.ml.catboost.CatboostRegressionModel)1 DummyVectorizer (org.apache.ignite.ml.dataset.feature.extractor.impl.DummyVectorizer)1 IgniteModelParser (org.apache.ignite.ml.inference.parser.IgniteModelParser)1 InMemoryModelReader (org.apache.ignite.ml.inference.reader.InMemoryModelReader)1