Updated to use ignite framework, simplifies a lot of stats

RicherMans / text_based_depression

Source code for the paper "Text-based Depression Detection: What Triggers An Alert"

44 stars 10 forks source link

Closed RicherMans closed 5 years ago

RicherMans commented 5 years ago

Enhanced loss to include a weighted average to enhance training stability. Removed tanh from attention, which further enhances stability.