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)
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)
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
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
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)
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)