Update README.md

README.md

@@ -13,62 +13,69 @@ widget:
   example_title: "Star Wars"
 
 ---
-Multi-Label Text classification model used to decode if passages contain a misfortunate event, a cause for misfortune, and/or an action to mollify or prevent some misfortune. The current F1 micro score of 140 passages not used for training is .678. individual class f1 scores shown below.
+Multi-Label Text classification model used to decode if passages contain a misfortunate event, a cause for misfortune, and/or an action to mollify or prevent some misfortune. Note that at this moment, some labels have been excluded as they are not relevant for the final use of the model
+8293 passages were used for Training and split into 5 folds (~6634 for train set, ~1659 for validation set over 5 folds). Although multiple parameters were tested, the current model uses:
+learning rate: 2e-5 
+weight decay: .01
+Dropout: .1
+Batch Size: 8
+Epochs: 15 (epoch 13 used here)
+The current F1 micro score of 2074 passages not used for training is .637. individual class f1 scores shown below. 
 <ul>
-  <li>EVENT:  .916
+  <li>EVENT:  -
     <ul>
       <li>
-        Illness:  .870
+        Illness:  .866
       </li>
       <li>
-        Accident:  .0
+        Accident:  .41
       </li>
       <li>
-        Other:  .0
+        Other:  .583
       </li>
     </ul>
   </li>
-  <li>CAUSE:  .863
+  <li>CAUSE:  -
     <ul>
       <li>
-        Just Happens:  .870
+        Just Happens:  -
       </li>
       <li>
-        Material Physical:  .0
+        Material Physical:  .431
       </li>
       <li>
-        Spirits and Gods:  .718
+        Spirits and Gods:  .667
       </li>
       <li>
-        Witchcraft and Sorcery:  .0
+        Witchcraft and Sorcery:  .615
       </li>
       <li>
-        Rule Violation Taboo:  .0
+        Rule Violation Taboo:  .555
       </li>
       <li>
-        Jealous Evil Eye:  .0
+        Jealous Evil Eye:  -
       </li>
     </ul>
   </li>
-  <li>ACTION:  .766
+  <li>ACTION:  -
     <ul>
       <li>
-        Physical Material:  .538
+        Physical Material:  .635
       </li>
       <li>
-        Technical Specialist:  .0
+        Technical Specialist:  .357
       </li>
       <li>
-        Divination:  .0
+        Divination:  .303
       </li>
       <li>
-        Shaman Medium Healer:  .08
+        Shaman Medium Healer:  .549
       </li>
       <li>
-        Priest High Religion:  .0
+        Priest High Religion:  .34
       </li>
       <li>
-        Other:  .0
+        Other:  -
       </li>
     </ul>
   </li>