search

RicherMans / text_based_depression

Source code for the paper "Text-based Depression Detection: What Triggers An Alert"
44 stars 10 forks source link

Using pretrained models #9

Closed ganfy closed 9 months ago

ganfy commented 9 months ago

Hi, I'm a little confused on the use of the pretrained models (from the models.zip) . Are they supposed to be used in the feature extraction, called from an extract_text_something file?

RicherMans commented 9 months ago

Hey, nope these are classifiers for each respective feature. The extract_text_x.py extracts only the features. These features need to be fed to the classifier.

ganfy commented 9 months ago

So I use them in the run.py, right?

RicherMans commented 9 months ago

Yes, First extract your features, then use run.py with evaluate and the respective model as input.

ganfy commented 9 months ago

I see, it works fine. Thank you.