TensorFlow-Keras मॉडल को .js (TensorFlow js) फॉर्मेट में कैसे सेव करें

0
SREERAG R NANDAN 20 पद 2019, 08:26

1 उत्तर

सबसे बढ़िया उत्तर

tensorflowjs पैकेज इंस्टॉल करने के बाद इस कोड को आजमाएं आपको ipython पैकेज को भी इंस्टॉल करना पड़ सकता है

pip install tensorflowjs pip install -U ipython

import tensorflowjs as tfjs
tfjs_target_dir = "./tfjs"
def train():
   model = keras.Sequential([
   keras.layers.Flatten(input_shape=(190, 190)),
   keras.layers.Dense(80, activation='relu'),
   keras.layers.Dense(2, activation='softmax')
  ])   # for example
   model.compile(optimizer='adam',
          loss='sparse_categorical_crossentropy',
          metrics=['accuracy']) # for example
   model.fit(train_data_images, train_data_labels, epochs=15) # for example
   tfjs.converters.save_keras_model(model, tfjs_target_dir)`
0
melledijkstra 7 जून 2020, 11:32