dragonSwing
commited on
Commit
•
4747a28
1
Parent(s):
ee09de4
Update README
Browse files
README.md
CHANGED
@@ -37,14 +37,14 @@ cache_dir = "./capu"
|
|
37 |
def download_files(repo_id, cache_dir=None, ignore_regex=None):
|
38 |
download_dir = snapshot_download(repo_id=repo_id, cache_dir=cache_dir, ignore_regex=ignore_regex)
|
39 |
if cache_dir is None or download_dir == cache_dir:
|
40 |
-
return download_dir
|
41 |
|
42 |
file_names = os.listdir(download_dir)
|
43 |
for file_name in file_names:
|
44 |
shutil.move(os.path.join(download_dir, file_name), cache_dir)
|
45 |
os.rmdir(download_dir)
|
46 |
return cache_dir
|
47 |
-
|
48 |
download_files(repo_id="dragonSwing/vibert-capu", cache_dir=cache_dir, ignore_regex=["*.json", "*.bin"])
|
49 |
sys.path.append(cache_dir)
|
50 |
```
|
@@ -74,11 +74,11 @@ Here is the number of product reviews we used for fine-tuning the model:
|
|
74 |
Below is a breakdown of the performance of the model by each label on 120,000 held-out text samples:
|
75 |
| label | precision | recall | f1-score | support|
|
76 |
| --------- | -------------|-------- | ----------|--------|
|
77 |
-
| **Upper** | 0.88 | 0.89 | 0.89 | 56497
|
78 |
-
| **Complex-Upper** | 0.92 | 0.83 | 0.88 | 480
|
79 |
-
| **.** | 0.81 | 0.82 | 0.82 | 18139
|
80 |
-
| **,** | 0.73 | 0.70 | 0.71 | 22961
|
81 |
-
| **:** | 0.74 | 0.56 | 0.64 | 1432
|
82 |
-
| **?** | 0.80 | 0.76 | 0.78 | 1730
|
83 |
-
| **none** | 0.99 | 0.99 | 0.99 |475611
|
84 |
-----------------------------------------------
|
|
|
37 |
def download_files(repo_id, cache_dir=None, ignore_regex=None):
|
38 |
download_dir = snapshot_download(repo_id=repo_id, cache_dir=cache_dir, ignore_regex=ignore_regex)
|
39 |
if cache_dir is None or download_dir == cache_dir:
|
40 |
+
return download_dir
|
41 |
|
42 |
file_names = os.listdir(download_dir)
|
43 |
for file_name in file_names:
|
44 |
shutil.move(os.path.join(download_dir, file_name), cache_dir)
|
45 |
os.rmdir(download_dir)
|
46 |
return cache_dir
|
47 |
+
|
48 |
download_files(repo_id="dragonSwing/vibert-capu", cache_dir=cache_dir, ignore_regex=["*.json", "*.bin"])
|
49 |
sys.path.append(cache_dir)
|
50 |
```
|
|
|
74 |
Below is a breakdown of the performance of the model by each label on 120,000 held-out text samples:
|
75 |
| label | precision | recall | f1-score | support|
|
76 |
| --------- | -------------|-------- | ----------|--------|
|
77 |
+
| **Upper** | 0.88 | 0.89 | 0.89 | 56497 |
|
78 |
+
| **Complex-Upper** | 0.92 | 0.83 | 0.88 | 480 |
|
79 |
+
| **.** | 0.81 | 0.82 | 0.82 | 18139 |
|
80 |
+
| **,** | 0.73 | 0.70 | 0.71 | 22961 |
|
81 |
+
| **:** | 0.74 | 0.56 | 0.64 | 1432 |
|
82 |
+
| **?** | 0.80 | 0.76 | 0.78 | 1730 |
|
83 |
+
| **none** | 0.99 | 0.99 | 0.99 |475611 |
|
84 |
-----------------------------------------------
|