tensorflow - 有没有从模型文件 ktrain 加载预测器的方法？

Question

我像这样保存了我的模型：

ktrain.get_predictor(learner.model,preproc=trans).save('model')

我想加载我的模型并使用它并执行以下操作：

predictor = ktrain.load(folder)
x = "hello wold"
prediction = predictor(x) 

现在，我有一个文件夹“模型”，其中包含 5 个文件：vocab.txt、tokenizer_config.json、tf_model.preproc、special_tokens_map.json 和 config.json 谢谢请帮我加载和使用我的预测器

Answer 1

predictor = ktrain.load_predictor('<Path>')
model = ktrain.get_predictor(predictor.model, predictor.preproc)
predictions = model.predict('<anything you want to predict')

There's also a method model.predict_filename() to predict from images.

Answer 2

您需要显式使用predictor.model.predict函数来进行预测。因此，流程：

predictor = ktrain.get_predictor(learner.model,preproc=trans).save('model')
predictor.save('model')

predictor = ktrain.load_predictor('model')
x = "Hello World"
prediction = predictor.model.predict(x)
print(prediction)