(cls)
| 36 | |
| 37 | @classmethod |
| 38 | def get_predictor_model(cls): |
| 39 | |
| 40 | # print(cls.searching_all_files(PATH)) |
| 41 | # Get model predictor |
| 42 | if cls.model is None: |
| 43 | with open(os.path.join(PATH, "model_config.json")) as f: |
| 44 | model_config = json.load(f) |
| 45 | |
| 46 | predictor = BertClassificationPredictor( |
| 47 | os.path.join(PATH, "model_out"), |
| 48 | label_path=PATH, |
| 49 | multi_label=bool(model_config["multi_label"]), |
| 50 | model_type=model_config["model_type"], |
| 51 | do_lower_case=bool(model_config["do_lower_case"]), |
| 52 | ) |
| 53 | cls.model = predictor |
| 54 | |
| 55 | return cls.model |
| 56 | |
| 57 | @classmethod |
| 58 | def predict(cls, text, bing_key=None): |
no test coverage detected