есть ли способ загрузить модель с тонкой настройкой?

Я сохранил свою модель вот так:

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

Я хочу загрузить свою модель и использовать ее, а также сделать что-то вроде:

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

теперь у меня есть папка «модель», которая содержит 6 файлов: vocab.txt, tokenizer_config.json, tf_model.preproc, special_tokens_map.json, model.h5 и config.json

но когда я запускаю свой код, который возвращает ошибку:

        File "api.py", line 11, in <module>
    predictor = ktrain.load_predictor("/model")
  File "C:\Users\omayma\AppData\Roaming\Python\Python38\site-packages\ktrain\core.py", line 1541, in load_predictor
    raise Exception('Failed to load .preproc file in either the post v0.16.x loction (%s) or pre v0.16.x location (%s)' % (os.path.join(fpath, U.PREPROC_NAME), fpath+'.preproc'))
Exception: Failed to load .preproc file in either the post v0.16.x loction (/model\tf_model.preproc) or pre v0.16.x location (/model.preproc)

0 ответов

Другие вопросы по тегам