text: from __future__ import print_function
  
  from pyspark import SparkContext
  
  import argparse
  
  from parallelm.mlops import mlops as pm
  from parallelm.mlops.e2e_tests import predict_node
  
  
  def parse_args():
  
      parser = argparse.ArgumentParser()
  
      parser.add_argument("--data-file", help="Data file to use as input")
      parser.add_argument("--input-model", help="Path of input model to use")
      parser.add_argument("--num-trees", type=int, default=3, help="Number of trees")
      parser.add_argument("--num-classes", type=int, default=2, help="Number of classes")
      parser.add_argument("--max-depth", type=int, default=4, help="Max depth to calculate")
      options = parser.parse_args()
  
      return options
  
  
  def main():
      options = parse_args()
  
      sc = SparkContext(appName="predict-test")
  
      pm.init(sc)
      predict_node(options)
      sc.stop()
      pm.done()
  
  
  if __name__ == "__main__":
      main()
  