helayers icon indicating copy to clipboard operation
helayers copied to clipboard
verified publisher icon IBM

Improve model performances on heart disease.

Open fd0r opened this issue 3 years ago • 2 comments

  • Change optimizer from SGD to Adam
  • Change loss from MSE to BinaryCrossEntropy(with_logits=True)

fd0r avatar Aug 22 '22 10:08 fd0r

With those changes the final test accuracy is improved from 53.333% to 93.333%.

fd0r avatar Aug 23 '22 14:08 fd0r

Thanks @fd0r for your contributions. We're looking into it.

dubek avatar Aug 23 '22 19:08 dubek