有没有从模型文件ktrain中加载predictor的方法？

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

还有一种方法model.predict_filename()可以从图像中进行预测。

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)