Update README.md

README.md

@@ -7,6 +7,7 @@ metrics:
 tags:
 - sentiment-analysis
 - social-listening
+library_name: transformers
 ---
 
 # 5CD-ViSoBERT for Vietnamese Sentiment Analysis
@@ -196,141 +197,148 @@ The model will give softmax outputs for three labels.
 <table>
         <tr align="center">
             <td rowspan=2><b>Model</td>
-            <td rowspan=2><b>Avg MF1</td>
-            <td colspan=3><b>SA-VLSP2016</td>
-            <td colspan=3><b>AIVIVN-2019</td>
-            <td colspan=3><b>UIT-VSFC</td>
-            <td colspan=3><b>UIT-VSMEC</td>
-            <td colspan=3><b>UIT-ViCTSD</td>
-            <td colspan=3><b>UIT-ViOCD</td>
-            <td colspan=3><b>UIT-ViSFD</td>
-            <td colspan=3><b>VOZ-HSD</td>
-            <td colspan=3><b>Vi-amazon-polar</td>
+            <td colspan=4><b>SA-VLSP2016</td>
+            <td colspan=4><b>AIVIVN-2019</td>
+            <td colspan=4><b>UIT-VSFC</td>
+            <td colspan=4><b>UIT-VSMEC</td>
+            <td colspan=4><b>UIT-ViCTSD</td>
         </tr>
         <tr align="center">
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
+        </tr>
+        <tr align="center">
+        <tr align="center">
+            <td align="left">wonrax/phobert-base-vietnamese-sentiment</td>
+            <td>61.65</td>
+            <td>63.95</td>
+            <td>61.65</td>
+            <td>60.01</td>
+            <td>84.87</td>
+            <td>95.12</td>
+            <td>84.87</td>
+            <td>89.47</td>
+            <td>76.37</td>
+            <td>88.10</td>
+            <td>76.37</td>
+            <td>79.53</td>
+            <td>65.41</td>
+            <td>74.36</td>
+            <td>65.41</td>
+            <td>68.33</td>
+            <td>62.34</td>
+            <td>73.08</td>
+            <td>62.34</td>
+            <td>65.54</td>
+        </tr>
+        <tr align="center">
+            <td align="left">5CD-AI/Vietnamese-sentiment-visobert-base</td>
+            <td><b>88.06</td>
+            <td><b>88.16</td>
+            <td><b>88.06</td>
+            <td><b>88.06</td>
+            <td><b>99.62</td>
+            <td><b>99.65</td>
+            <td><b>99.62</td>
+            <td><b>99.64</td>
+            <td><b>94.65</td>
+            <td><b>93.30</td>
+            <td><b>93.65</td>
+            <td><b>93.38</td>
+            <td><b>77.91</td>
+            <td><b>77.21</td>
+            <td><b>77.91</td>
+            <td><b>77.46</td>
+            <td><b>75.10</td>
+            <td><b>74.59</td>
+            <td><b>75.10</td>
+            <td><b>74.79</td>
+        </tr>
+    </div>
+</table>
+
+
+<table>
+        <tr align="center">
+            <td rowspan=2><b>Model</td>
+            <td colspan=4><b>UIT-ViOCD</td>
+            <td colspan=4><b>UIT-ViSFD</td>
+            <td colspan=4><b>VOZ-HSD</td>
+            <td colspan=4><b>Vi-amazon-polar</td>
+        </tr>
+        <tr align="center">
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
             <td><b>Acc</td>
+            <td><b>Prec</td>
+            <td><b>Recall</td>
             <td><b>WF1</td>
-            <td><b>MF1</td>
-        </tr>
-        <tr align="center">
-            <td align="left">viBERT</td>
-            <td>78.16</td>
-            <td>61.91</td>
-            <td>61.98</td>
-            <td>59.7</td>
-            <td>85.34</td>
-            <td>85.01</td>
-            <td>62.07</td>
-            <td>89.93</td>
-            <td>89.79</td>
-            <td>76.8</td>
-            <td>90.42</td>
-            <td>90.45</td>
-            <td>84.55</td>
         </tr>
         <tr align="center">
-            <td align="left">vELECTRA</td>
-            <td>79.23</td>
-            <td>64.79</td>
-            <td>64.71</td>
-            <td>61.95</td>
-            <td>86.96</td>
-            <td>86.37</td>
-            <td>63.95</td>
-            <td>89.83</td>
-            <td>89.68</td>
-            <td>76.23</td>
-            <td>90.59</td>
-            <td>90.58</td>
-            <td>85.12</td>
-        </tr>
-        <tr align="center">
-            <td align="left">PhoBERT-Base </td>
-            <td>79.3</td>
-            <td>63.49</td>
-            <td>63.36</td>
-            <td>61.41</td>
-            <td>87.12</td>
-            <td>86.81</td>
-            <td>65.01</td>
-            <td>89.83</td>
-            <td>89.75</td>
-            <td>76.18</td>
-            <td>91.32</td>
-            <td>91.38</td>
-            <td>85.92</td>
-        </tr>
-        <tr align="center">
-            <td align="left">PhoBERT-Large</td>
-            <td>79.82</td>
-            <td>64.71</td>
-            <td>64.66</td>
-            <td>62.55</td>
-            <td>87.32</td>
-            <td>86.98</td>
-            <td>65.14</td>
-            <td>90.12</td>
-            <td>90.03</td>
-            <td>76.88</td>
-            <td>91.44</td>
-            <td>91.46</td>
-            <td>86.56</td>
-        </tr>
         <tr align="center">
-            <td align="left">ViSoBERT</td>
-            <td>81.58</td>
-            <td>68.1</td>
-            <td>68.37</td>
-            <td>65.88</td>
-            <td>88.51</td>
-            <td>88.31</td>
-            <td>68.77</td>
-            <td>90.99</td>
-            <td><b>90.92</td>
-            <td><b>79.06</td>
-            <td>91.62</td>
-            <td>91.57</td>
-            <td>86.8</td>
+            <td align="left">wonrax/phobert-base-vietnamese-sentiment</td>
+            <td>87.14</td>
+            <td>74.68</td>
+            <td>78.13</td>
+            <td>67.90</td>
+            <td>67.95</td>
+            <td>67.90</td>
+            <td>66.98</td>
+            <td>51.89</td>
+            <td>60.18</td>
+            <td>51.89</td>
+            <td>53.61</td>
+            <td>61.40</td>
+            <td>76.53</td>
+            <td>61.40</td>
+            <td>65.70</td>
         </tr>
         <tr align="center">
-            <td align="left">visobert-14gb-corpus</td>
-            <td><b>82.2</td>
-            <td><b>68.69</td>
-            <td><b>68.75</td>
-            <td><b>66.03</td>
-            <td><b>88.79</td>
-            <td><b>88.6</td>
-            <td><b>69.57</td>
-            <td><b>91.02</td>
-            <td>90.88</td>
-            <td>77.13</td>
-            <td><b>93.69</td>
-            <td><b>93.63</td>
-            <td><b>89.66</td>
+            <td align="left">5CD-AI/Vietnamese-sentiment-visobert-base</td>
+            <td><b>94.35</td>
+            <td><b>94.74</td>
+            <td><b>94.35</td>
+            <td><b>94.53</td>
+            <td><b>93.26</td>
+            <td><b>93.20</td>
+            <td><b>93.26</td>
+            <td><b>93.21</td>
+            <td><b>67.78</td>
+            <td><b>69.82</td>
+            <td><b>67.78</td>
+            <td><b>68.39</td>
+            <td><b>89.90</td>
+            <td><b>90.13</td>
+            <td><b>89.90</td>
+            <td><b>90.01</td>
         </tr>
     </div>
 </table>