How to save/load GraphSAGE model with keras ?

[HripsimeS]

Hello @sachinprasadhs @SuryanarayanaY

Please find the python script below. To run the part 1, it seems the model in baseline is missing (the name "model" is not defined). https://github.com/Yaxin-Cui/Graph-Neural-Network/blob/main/code/graphstellar_updated.py

To try fix this issue, I tried to save the trained model in the part 0 with two options to see which one works efficient. model.save("model.keras") or model = keras.models.load_model("model.h5")

And after loading the model in part 1 with these lines: model = keras.models.load_model("model.keras") or model = keras.models.load_model("model.h5")

In case of h5 format I received the following error

And in case of keras format I received the following error

Can you please help to see how to fix the issues or maybe if there is other way to save this model, which one can be the best.