问题描述
我像这样保存了我的模型:
ktrain.get_predictor(learner.model,preproc=trans).save('model')
我想加载我的模型并使用它并执行以下操作:
predictor = ktrain.load(folder)
x = "hello wold"
prediction = predictor(x)
现在,我有一个文件夹“model”,其中包含 5 个文件:vocab.txt、tokenizer_config.json、tf_model.preproc、special_tokens_map.json 和 config.json 谢谢,请帮我加载和使用我的预测器
解决方法
predictor = ktrain.load_predictor('<Path>')
model = ktrain.get_predictor(predictor.model,predictor.preproc)
predictions = model.predict('<anything you want to predict')
还有一种方法 model.predict_filename()
可以根据图像进行预测。