dragonSwing commited on
Commit
4747a28
1 Parent(s): ee09de4

Update README

Browse files
Files changed (1) hide show
  1. README.md +9 -9
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
  -----------------------------------------------