fenglinliu commited on
Commit
f216670
·
verified ·
1 Parent(s): 616c89c

Update gen_table.py

Browse files
Files changed (1) hide show
  1. gen_table.py +16 -7
gen_table.py CHANGED
@@ -34,18 +34,27 @@ def model_size_flag(sz, FIELDS):
34
  return False
35
 
36
 
 
 
 
 
 
 
 
 
 
 
 
 
37
  def model_type_flag(line, FIELDS):
38
- if 'Commercial LLMs' in FIELDS and line['Commercial LLMs'] == 'Yes':
39
- return True
40
- if 'General LLMs' in FIELDS and line['General LLMs'] == 'Yes':
41
  return True
42
- if 'Medical LLMs' in FIELDS and line['Medical LLMs'] == 'Yes':
43
  return True
44
- if 'SOTA' in FIELDS and line['SOTA'] == 'Yes':
45
  return True
46
  return False
47
-
48
-
49
  def BUILD_L1_DF(results, fields):
50
  check_box = {}
51
  check_box['essential'] = ['Method', 'Param (B)']
 
34
  return False
35
 
36
 
37
+ # def model_type_flag(line, FIELDS):
38
+ # if 'Commercial LLMs' in FIELDS and line['Commercial LLMs'] == 'Yes':
39
+ # return True
40
+ # if 'General LLMs' in FIELDS and line['General LLMs'] == 'Yes':
41
+ # return True
42
+ # if 'Medical LLMs' in FIELDS and line['Medical LLMs'] == 'Yes':
43
+ # return True
44
+ # if 'SOTA' in FIELDS and line['SOTA'] == 'Yes':
45
+ # return True
46
+ # return False
47
+
48
+
49
  def model_type_flag(line, FIELDS):
50
+ if 'OpenSource' in FIELDS and line['OpenSource'] == 'Yes':
 
 
51
  return True
52
+ if 'API' in FIELDS and line['OpenSource'] == 'No' and line['Verified'] == 'Yes':
53
  return True
54
+ if 'Proprietary' in FIELDS and line['OpenSource'] == 'No' and line['Verified'] == 'No':
55
  return True
56
  return False
57
+
 
58
  def BUILD_L1_DF(results, fields):
59
  check_box = {}
60
  check_box['essential'] = ['Method', 'Param (B)']