예제 #1
0
파일: utils_test.py 프로젝트: lyqsr/hwrt
def evaluate_model_single_recording_preloaded_test():
    """Test if the packaged model can be used."""
    model_path = pkg_resources.resource_filename('hwrt', 'misc/')
    model_file = os.path.join(model_path, "model.tar")
    recording = [[{'x': 12, 'y': 42, 'time': 123}]]
    utils.evaluate_model_single_recording(model_file, json.dumps(recording))
    (pq, fl, model, output_semantics) = utils.load_model(model_file)
예제 #2
0
파일: utils_test.py 프로젝트: shalevy1/hwrt
def test_evaluate_model_single_recording_preloaded():
    """Test if the packaged model can be used."""
    model_path = pkg_resources.resource_filename("hwrt", "misc/")
    model_file = os.path.join(model_path, "model.tar")
    recording = [[{"x": 12, "y": 42, "time": 123}]]
    utils.evaluate_model_single_recording(model_file, json.dumps(recording))
    (pq, fl, model, output_semantics) = utils.load_model(model_file)
예제 #3
0
파일: utils_test.py 프로젝트: Duum/hwrt
def evaluate_model_single_recording_preloaded_test():
    """Test if the packaged model can be used."""
    model_path = pkg_resources.resource_filename('hwrt', 'misc/')
    model_file = os.path.join(model_path, "model.tar")
    recording = [[{'x': 12, 'y': 42, 'time': 123}]]
    utils.evaluate_model_single_recording(model_file, json.dumps(recording))
    (pq, fl, model, output_semantics) = utils.load_model(model_file)
예제 #4
0
파일: classify.py 프로젝트: lyqsr/hwrt
 def __init__(self):
     logging.info("Start reading model...")
     model_path = pkg_resources.resource_filename('hwrt', 'misc/')
     model_file = os.path.join(model_path, "model.tar")
     logging.info("Model: %s", model_file)
     (preprocessing_queue, feature_list, model,
      output_semantics) = utils.load_model(model_file)
     self.preprocessing_queue = preprocessing_queue
     self.feature_list = feature_list
     self.model = model
     self.output_semantics_orig = output_semantics
     self.output_semantics = self.output_semantics_orig
     if 'HWRT_ENV' in os.environ and os.environ['HWRT_ENV'] == 'dev':
         new_semantics = []
         for el in output_semantics:
             new_semantics.append(el.split(";")[1])
         self.output_semantics = new_semantics
예제 #5
0
파일: classify.py 프로젝트: Duum/hwrt
 def __init__(self):
     logging.info("Start reading model...")
     model_path = pkg_resources.resource_filename('hwrt', 'misc/')
     model_file = os.path.join(model_path, "model.tar")
     logging.info("Model: %s", model_file)
     (preprocessing_queue, feature_list, model,
      output_semantics) = utils.load_model(model_file)
     self.preprocessing_queue = preprocessing_queue
     self.feature_list = feature_list
     self.model = model
     self.output_semantics_orig = output_semantics
     self.output_semantics = self.output_semantics_orig
     if 'HWRT_ENV' in os.environ and os.environ['HWRT_ENV'] == 'dev':
         new_semantics = []
         for el in output_semantics:
             new_semantics.append(el.split(";")[1])
         self.output_semantics = new_semantics
예제 #6
0
파일: classify.py 프로젝트: shalevy1/hwrt
 def __init__(self):
     logger.info("Start reading model...")
     model_path = pkg_resources.resource_filename(__name__, "misc/")
     model_file = os.path.join(model_path, "model.tar")
     logger.info(f"Model: {model_file}")
     (preprocessing_queue, feature_list, model,
      output_semantics) = utils.load_model(model_file)
     self.preprocessing_queue = preprocessing_queue
     self.feature_list = feature_list
     self.model = model
     self.output_semantics_orig = output_semantics
     self.output_semantics = self.output_semantics_orig
     if "HWRT_ENV" in os.environ and os.environ["HWRT_ENV"] == "dev":
         new_semantics = []
         for el in output_semantics:
             new_semantics.append(el.split(";")[1])
         self.output_semantics = new_semantics
예제 #7
0
파일: utils_test.py 프로젝트: shalevy1/hwrt
def test_load_model():
    """Test if the packaged model can be loaded."""
    model_path = pkg_resources.resource_filename("hwrt", "misc/")
    model_file = os.path.join(model_path, "model.tar")
    utils.load_model(model_file)
예제 #8
0
파일: utils_test.py 프로젝트: Duum/hwrt
def load_model_test():
    """Test if the packaged model can be loaded."""
    model_path = pkg_resources.resource_filename('hwrt', 'misc/')
    model_file = os.path.join(model_path, "model.tar")
    utils.load_model(model_file)
예제 #9
0
파일: utils_test.py 프로젝트: lyqsr/hwrt
def load_model_test():
    """Test if the packaged model can be loaded."""
    model_path = pkg_resources.resource_filename('hwrt', 'misc/')
    model_file = os.path.join(model_path, "model.tar")
    utils.load_model(model_file)