Model de/serialization

Lockdain · Feb 14, 2023 · 82eb19b · 82eb19b
1 parent 8bdbc99
commit 82eb19b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -5,7 +5,7 @@
     <modelVersion>4.0.0</modelVersion>
 
     <groupId>ru.asergeenko</groupId>
-    <artifactId>Alink Sentiment Analysis</artifactId>
+    <artifactId>AlinkSentiment</artifactId>
     <version>1.0-SNAPSHOT</version>
 
     <properties>

diff --git a/src/main/java/ru/asergeenko/flink/ml/iris/App.java b/src/main/java/ru/asergeenko/flink/ml/iris/App.java
@@ -48,7 +48,9 @@ public static void main(String[] args) throws Exception {
         );
 
         PipelineModel model = pipeline.fit(trainingSet);
-        model.transform(validationSet.firstN(100))
+        model.save("src/main/resources/model");
+        PipelineModel modelDes = PipelineModel.load("src/main/resources/model");
+        modelDes.transform(validationSet.firstN(100))
                 .select(new String[]{"text", "label", "pred"})
                 .firstN(5)
                 .print();