emonet.get_weights module#

Module to calculate emotion weights per therapist and export to JSON file.

Output can be read into loss functions for weighting during classification.

emonet.get_weights.get_weights(therapist: str, emotion: str, split: str = 'train', binary: bool = True) → Dict[source]#

Get therapist emotion rates in order of increasing severity.

Parameters

  • therapist (str) – Therapist weights to retrieve.

  • emotion (str) – Emotion to retrieve.

  • split (str) – Split to retreive

  • binary (bool) – Indicator to retrieve binary yes/no labels or original none/low/medium/high labels.

Returns

Dict – Dictionary record including therapist, emotion, counts and weights per.