google-research
ValueError raised upon calling RetVecTokenizer function
Hi!
I am trying to use RetVec as an embedding layer for a email spam classification project. When the RetVecTokenizer function is called, I get a Value Error which is described below:
File format not supported: filepath=/root/.keras/retvec-v1. Keras 3 only supports V3 .keras files and legacy H5 format files (.h5 extension). Note that the legacy SavedModel format is not supported by load_model() in Keras 3. In order to reload a TensorFlow SavedModel as an inference-only layer in Keras 3, use keras.layers.TFSMLayer(/root/.keras/retvec-v1, call_endpoint='serving_default') (note that your call_endpoint might have a different name).
Here is the code I wrote:
inputs = layers.Input(shape=(1, ), name="token", dtype=tf.string)
lstm_1 = tf.keras.layers.LSTM(20, dropout=0.2, return_sequences=True)(x) lstm_2 = tf.keras.layers.LSTM(20, dropout=0.2, return_sequences=True)(lstm_1) flatten = tf.keras.layers.Flatten()(lstm_2) dropout = tf.keras.layers.Dropout(0.2, name="dropout")(flatten) dense = tf.keras.layers.Dense(1, activation="sigmoid")(dropout)
model = tf.keras.Model(inputs=[inputs], outputs=[dense])
How to resolve this?
Addendum: 2.7.4 can train the model but cannot execute the prediction code, with 2.8.4 you can train as well as execute the prediction.