Блог

Главная - Вопросы по программированию - У меня есть модель keras с Conv1D, а до этого-кодировщик этикеток, один горячий кодировщик и токенизатор. Я хочу сохранить модель с помощью конвейера и использовать

У меня есть модель keras с Conv1D, а до этого-кодировщик этикеток, один горячий кодировщик и токенизатор. Я хочу сохранить модель с помощью конвейера и использовать

#tensorflow #keras #neural-network #pipeline #tensorflow-serving

Вопрос:

Шаги, которые я выполняю,

  1. Кодер этикеток
  2. одно горячее кодирование с использованием tf.keras.utils.to_categorical
  3. маркировщик текста keras pp —gt; маркировщик = Tokenizer(num_words = vocab_size, oov_token=oov_tok)
  4. Последовательная модель с встраиванием, Conv1D, GlobalMaxPooling1D, 2 плотных слоя.

Установил модель на тренировке и проверил все параметры. Теперь я хотел сохранить модель. Кто-то предложил использовать трубопровод там, я могу поместить все вышеперечисленные 4 вещи и создать трубопровод, а затем использовать в пункте назначения для прогнозирования с использованием того же трубопровода.

Кто-нибудь может мне помочь, как создать конвейер, сохранить и загрузить модель?

Метки: , ,