Do you have a dataset containing the seizure intensities in [0, inf]?
If so, it seems like a regression use case for me, i.e. you could use a linear layer (maybe with a relu at the end) and something like nn.MSELoss as your criterion.
Could you explain the stopping condition a bit?