есть ли способ загрузить предиктор из файла модели 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.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)

Answer 2

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

Также есть метод model.predict_filename() предсказывать по изображениям.