(self, experiment_name, version_name)
| 39 | |
| 40 | class TextMatchWrapper(mlflow.pyfunc.PythonModel): |
| 41 | def __init__(self, experiment_name, version_name): |
| 42 | self.experiment_name = experiment_name |
| 43 | self.version_name = version_name |
| 44 | self.wordstest_dict = {"0":"我去玉龙雪山并且喜欢玉龙雪山玉龙雪山", "1":"我在玉龙雪山并且喜欢玉龙雪山", "2":"我在九寨沟", "3":"你好"} #["我去玉龙雪山并且喜欢玉龙雪山玉龙雪山","我在玉龙雪山并且喜欢玉龙雪山","我在九寨沟"] |
| 45 | self.mf = ModelFactory( match_models=['bow', 'tfidf', 'ngram_tfidf'] ) |
| 46 | |
| 47 | def load_context(self, context): |
| 48 | # wordstest_dict = context.artifacts["wordstest_dict"] |
nothing calls this directly
no test coverage detected