Hi there! Very minor spelling error I noticed that in the example notebook provided the attribute list is provided as:
attributes = [ '**antagonize**' , 'condescending', 'dismissive', 'generalisation', 'generalisation_unfair', 'healthy', 'hostile', 'sarcastic']
However, train.csv and unhealthy_full.csv have the headings as:
**antagonise**,condescending,dismissive,generalisation,generalisation_unfair,healthy,hostile,sarcastic
This leads to an indexing error when running the notebook e.g. antagonize is not a valid index for the training data. Just a heads up for other users