def reset_states()

in train/CustomModel.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

    def reset_states(self):
        self.tp.assign(0)
        self.trues.assign(0)