Notebook Sixteen | Repository

Personality Traits

Andrea Leone
University of Trento
February 2022


Load BART pre-trained NLI model for zero-shot classification and set up the classification pipeline


Organise the personality descriptors for the "Big Five" traits and store traits and indices for the classifier


Set up the logic to tidy the single classification scores, aggregate them, and compute the trait results


Set up the logic to plot the results


Get a sample transcription


Test the model on the sample text


Test on a couple of blog posts of David Heinemeier Hansson


Test on the YouTube transcripts of ArkInvest's Cathie Wood podcast


Capture personality traits for each talk in the dataset