def call(self, x):
            """
            Call function to evaluate the model

            :param x: Input to the model
            :return: Prediction of the model
            """
            x = self.dense1(x)
            x = self.dense2(x)
            x = self.logits(x)
            return x

            Call function to evaluate the model

            :param x: Input to the model
            :return: Prediction of the model
            
tests/utils.py