Open vijender412 opened 3 years ago
The plotting of Keras models created with the Models API
(by subclassing the Model class, in other words) does not work out of the box.
You can find an existing discussion and some workarounds (which I haven't tried, and I don't know if they work) here: https://stackoverflow.com/questions/61427583/how-do-i-plot-a-keras-tensorflow-subclassing-api-model
Hi Everyone, I am new to tensorflow and tensorflow recommenders. I have tried DCN and other architectures. When I am trying to plot using
tf.keras.utils.plot_model(model, "mymodel.png", show_shapes=False)
It only shows model name not the complete diagram.I am trying below code `class DCN(tfrs.Model):
def init(self, use_cross_layer, deep_layer_sizes, projection_dim=None): super().init()
def call(self, features):
Concatenate embeddings
def compute_loss(self, features, training=False): labels = features.pop("user_rating") scores = self(features) return self.task( labels=labels, predictions=scores, )`
Can anyone help in this. I am able to see summary but not the diagram.