Start the h2o and pmml server if possible

Author: tobegit3hub

requirements.txt

@@ -8,6 +8,6 @@ tensorflow>=1.0.0
 scikit-learn
 xgboost
 mxnet
-h2o
+h2o==3.18.0.8
 onnx
 #git+https://github.com/openscoring/openscoring-python.git

simple_tensorflow_serving/h2o_inference_service.py

@@ -2,9 +2,11 @@
 from __future__ import division
 from __future__ import print_function
 
+import os
 import logging
 import time
 import json
+import subprocess
 
 from .abstract_inference_service import AbstractInferenceService
 from . import filesystem_util
@@ -29,6 +31,14 @@ def __init__(self, model_name, model_base_path):
     """
     super(H2oInferenceService, self).__init__()
 
+    # Start the h2o server
+    if os.path.isfile("/tmp/h2o.jar"):
+      logging.info("Run to run command 'java -jar /tmp/h2o.jar'")
+      subprocess.Popen(["java", "-jar", "/tmp/h2o.jar"])
+
+      logging.info("Sleep 10s to wait for h2o server")
+      time.sleep(10)
+
     local_model_base_path = filesystem_util.download_hdfs_moels(
         model_base_path)
 

simple_tensorflow_serving/pmml_inference_service.py

@@ -2,8 +2,10 @@
 from __future__ import division
 from __future__ import print_function
 
+import os
 import logging
 import time
+import subprocess
 
 from .abstract_inference_service import AbstractInferenceService
 from . import filesystem_util
@@ -29,6 +31,14 @@ def __init__(self, model_name, model_base_path):
 
     super(PmmlInferenceService, self).__init__()
 
+    # Start the pmml server
+    if os.path.isfile("/tmp/openscoring-server-executable-1.4-SNAPSHOT.jar"):
+      logging.info("Run to run command 'java -jar /tmp/openscoring-server-executable-1.4-SNAPSHOT.jar'")
+      subprocess.Popen(["java", "-jar", "/tmp/openscoring-server-executable-1.4-SNAPSHOT.jar"])
+
+      logging.info("Sleep 10s to wait for pmml server")
+      time.sleep(10)
+
     local_model_base_path = filesystem_util.download_hdfs_moels(
         model_base_path)
 

third_party/h2o/README.md

@@ -0,0 +1,5 @@
+# H2O
+
+## Usage
+
+Download with `wget http://h2o-release.s3.amazonaws.com/h2o/rel-wolpert/8/h2o-3.18.0.8.zip`.