[D] How to implement Last layer Laplace in tensorflow

Hi,

I am doing some multi label text classification in Tensorflow using sentence encoders. As my model is over confident while I am searching for solution, I found this url: .[https://jramkiss.github.io/2020/07/29/overconfident-nn/](https://jramkiss.github.io/2020/07/29/overconfident-nn/).

basically what article is saying " **The proposed solution is to only have a posterior over the weights in the last layer** " . The original implementation is in torch

I have idea to get the weights of the last layer like this

custom_model.get_layer('last_layer').weights

But I am facing challenges …

machinelearning tensorflow