Skip to content
Sam-Nielsen-Dot edited this page May 30, 2021 · 5 revisions

Welcome to the depressionAnalysis wiki!


Classification

classify(text, mode="string", switchpoint=0.95, model_id=31, classifier=None)

Classifies a string as either "Positive" or "Negative for depression.

Modes

  • string - returns output as either "Positive" or "Negative"
  • int - returns output as either 1 for positive or 0 for negative
  • probabilities - returns a list of tuples containing the value and the associated probability

Switchpoint

A value between 0 and 1 above which the positive probability must be for it to register as positive

Model id

Id of the model used to classify the text. An int between 0 and 95

Refer to the chart below for the various datasets and accuracies each model was trained on. 31 is the default model as it performs consistently higher in real world application'

Classifier

Provide a classifier object using the get_classifier(id). This saves on load time and should be used if classifying multiple items with the same model id.

Clone this wiki locally