Ошибка tf.keras.experimental.export_saved_model: __init __ () отсутствует 1 обязательный позиционный аргумент: 'feature_columns'

Question

Я пытаюсь сохранить свою модель в формате .pb или .pbtxt, используя tf.keras.experimental.export_saved_model (model, save_model_dir), как описано в следующем руководстве:

https://github.com/GoogleCloudPlatform/data-science-on-gcp/blob/master/updates/cloudml/flights_model_tf2.ipynb

но я получаю ошибку:

init () отсутствует 1 обязательный позиционный аргумент: 'feature_columns'

Заранее благодарю за помощь. Код ниже:

#my code, the model trains without any problems:

training_history = model.fit(train_ds, 
                    validation_data=val_ds,
                    epochs=epochs, 
                    callbacks=[EarlyStopping(patience=patience), tensorboard_callback ])

now=str(datetime.datetime.now()).replace(':','_').replace(' ','_').replace('.','_')

saved_model_dir="saved_models/"+now

tf.keras.experimental.export_saved_model(model, saved_model_dir)

#I get the following error:

TypeError: __init__() missing 1 required positional argument: 'feature_columns'