edichief's picture
Update spaCy pipeline
c40dc5b
{
"lang":"en",
"name":"textcat_goemotions",
"version":"0.0.1",
"description":"# \ud83e\ude90 spaCy Project: Categorization of emotions in Reddit posts (Text Classification) This project uses spaCy to train a text classifier on the [GoEmotions dataset](https://github.com/google-research/google-research/tree/master/goemotions)",
"author":"Explosion",
"email":"",
"url":"explosion.ai",
"license":"MIT",
"spacy_version":">=3.1.1,<3.2.0",
"spacy_git_version":"530b5d72f",
"vectors":{
"width":0,
"vectors":0,
"keys":0,
"name":null
},
"labels":{
"transformer":[
],
"textcat_multilabel":[
"admiration",
"amusement",
"anger",
"annoyance",
"approval",
"caring",
"confusion",
"curiosity",
"desire",
"disappointment",
"disapproval",
"disgust",
"embarrassment",
"excitement",
"fear",
"gratitude",
"grief",
"joy",
"love",
"nervousness",
"optimism",
"pride",
"realization",
"relief",
"remorse",
"sadness",
"surprise",
"neutral"
]
},
"pipeline":[
"transformer",
"textcat_multilabel"
],
"components":[
"transformer",
"textcat_multilabel"
],
"disabled":[
],
"_sourced_vectors_hashes":{
},
"performance":{
"cats_score":0.9022125863,
"cats_score_desc":"macro AUC",
"cats_micro_p":0.6666666667,
"cats_micro_r":0.4780564263,
"cats_micro_f":0.5568233683,
"cats_macro_p":0.550045931,
"cats_macro_r":0.419275998,
"cats_macro_f":0.4629156981,
"cats_macro_auc":0.9022125863,
"cats_f_per_type":{
"admiration":{
"p":0.7392197125,
"r":0.737704918,
"f":0.7384615385
},
"amusement":{
"p":0.7814569536,
"r":0.7788778878,
"f":0.7801652893
},
"anger":{
"p":0.5531914894,
"r":0.4,
"f":0.4642857143
},
"annoyance":{
"p":0.4,
"r":0.3564356436,
"f":0.3769633508
},
"approval":{
"p":0.4594594595,
"r":0.2569269521,
"f":0.3295638126
},
"caring":{
"p":0.5531914894,
"r":0.339869281,
"f":0.4210526316
},
"confusion":{
"p":0.6086956522,
"r":0.2763157895,
"f":0.3800904977
},
"curiosity":{
"p":0.5206185567,
"r":0.4072580645,
"f":0.4570135747
},
"desire":{
"p":0.5487804878,
"r":0.5844155844,
"f":0.5660377358
},
"disappointment":{
"p":0.5862068966,
"r":0.1042944785,
"f":0.1770833333
},
"disapproval":{
"p":0.5071428571,
"r":0.2431506849,
"f":0.3287037037
},
"disgust":{
"p":0.6904761905,
"r":0.2989690722,
"f":0.4172661871
},
"embarrassment":{
"p":0.7391304348,
"r":0.4857142857,
"f":0.5862068966
},
"excitement":{
"p":0.5652173913,
"r":0.2708333333,
"f":0.3661971831
},
"fear":{
"p":0.6555555556,
"r":0.6555555556,
"f":0.6555555556
},
"gratitude":{
"p":0.9459459459,
"r":0.8798882682,
"f":0.9117221418
},
"grief":{
"p":0.0,
"r":0.0,
"f":0.0
},
"joy":{
"p":0.6115107914,
"r":0.4941860465,
"f":0.5466237942
},
"love":{
"p":0.7446808511,
"r":0.8333333333,
"f":0.7865168539
},
"nervousness":{
"p":0.6428571429,
"r":0.4285714286,
"f":0.5142857143
},
"optimism":{
"p":0.6725146199,
"r":0.5502392344,
"f":0.6052631579
},
"pride":{
"p":0.0,
"r":0.0,
"f":0.0
},
"realization":{
"p":0.2978723404,
"r":0.2204724409,
"f":0.2533936652
},
"relief":{
"p":0.0,
"r":0.0,
"f":0.0
},
"remorse":{
"p":0.6849315068,
"r":0.7352941176,
"f":0.7092198582
},
"sadness":{
"p":0.5,
"r":0.4965034965,
"f":0.498245614
},
"surprise":{
"p":0.6288659794,
"r":0.4728682171,
"f":0.5398230088
},
"neutral":{
"p":0.7637637638,
"r":0.4320498301,
"f":0.5518987342
}
},
"cats_macro_auc_per_type":0.0,
"transformer_loss":0.8351497036,
"textcat_multilabel_loss":45.4983812743
},
"requirements":[
"spacy_transformers>=1.0.1,<1.1.0"
]
}