Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
@@ -12,7 +12,7 @@ ALL_UG40_LANGUAGES = ["ach", "eng", "lgg", "lug", "nyn", "rny", "teo", "swa"]
|
|
12 |
|
13 |
LANGUAGE_NAMES = {
|
14 |
"ach": "Acholi",
|
15 |
-
"eng": "English",
|
16 |
"lgg": "Lugbara",
|
17 |
"lug": "Luganda",
|
18 |
"nyn": "Runyankole",
|
@@ -24,34 +24,19 @@ LANGUAGE_NAMES = {
|
|
24 |
# Google Translate supported subset (for fair comparison)
|
25 |
GOOGLE_SUPPORTED_LANGUAGES = ["lug", "ach", "swa", "eng"]
|
26 |
|
27 |
-
#
|
28 |
-
GOOGLE_LANG_MAP = {"lug": "lg", "ach": "ach", "swa": "sw", "eng": "en"}
|
29 |
-
|
30 |
-
# SCIENTIFIC EVALUATION TRACKS
|
31 |
EVALUATION_TRACKS = {
|
32 |
"google_comparable": {
|
33 |
"name": "Google-Comparable Track",
|
34 |
-
"description": "Models evaluated
|
35 |
"languages": GOOGLE_SUPPORTED_LANGUAGES,
|
36 |
"min_samples_per_pair": 50,
|
37 |
-
"statistical_power": 0.8,
|
38 |
-
"significance_level": 0.05,
|
39 |
},
|
40 |
"ug40_complete": {
|
41 |
-
"name": "UG40-Complete Track",
|
42 |
-
"description": "Models evaluated on all UG40 language pairs",
|
43 |
"languages": ALL_UG40_LANGUAGES,
|
44 |
"min_samples_per_pair": 30,
|
45 |
-
"statistical_power": 0.8,
|
46 |
-
"significance_level": 0.05,
|
47 |
-
},
|
48 |
-
"language_pair_matrix": {
|
49 |
-
"name": "Language-Pair Matrix",
|
50 |
-
"description": "Individual language pair analysis with statistical significance",
|
51 |
-
"languages": ALL_UG40_LANGUAGES,
|
52 |
-
"min_samples_per_pair": 20,
|
53 |
-
"statistical_power": 0.7,
|
54 |
-
"significance_level": 0.05,
|
55 |
},
|
56 |
}
|
57 |
|
@@ -64,7 +49,7 @@ MODEL_CATEGORIES = {
|
|
64 |
"color": "#1f77b4",
|
65 |
},
|
66 |
"research": {
|
67 |
-
"name": "Research Models",
|
68 |
"description": "Academic and research institution models",
|
69 |
"examples": ["nllb", "m2m100"],
|
70 |
"color": "#ff7f0e",
|
@@ -77,46 +62,20 @@ MODEL_CATEGORIES = {
|
|
77 |
},
|
78 |
"community": {
|
79 |
"name": "Community Submissions",
|
80 |
-
"description": "User-submitted models and fine-tuned variants",
|
81 |
"examples": ["user_submission"],
|
82 |
"color": "#d62728",
|
83 |
},
|
84 |
}
|
85 |
|
86 |
-
#
|
87 |
-
STATISTICAL_CONFIG = {
|
88 |
-
"confidence_level": 0.95,
|
89 |
-
"bootstrap_samples": 1000,
|
90 |
-
"min_samples_for_ci": 20,
|
91 |
-
"effect_size_thresholds": {
|
92 |
-
"small": 0.2,
|
93 |
-
"medium": 0.5,
|
94 |
-
"large": 0.8,
|
95 |
-
},
|
96 |
-
"multiple_testing_correction": "bonferroni",
|
97 |
-
"outlier_detection": {
|
98 |
-
"method": "iqr",
|
99 |
-
"factor": 1.5,
|
100 |
-
},
|
101 |
-
}
|
102 |
-
|
103 |
-
# METRICS CONFIGURATION - Enhanced for statistical analysis
|
104 |
METRICS_CONFIG = {
|
105 |
"primary_metrics": ["bleu", "chrf", "quality_score"],
|
106 |
-
"secondary_metrics": ["rouge1", "
|
107 |
"display_precision": 4,
|
108 |
-
"
|
109 |
-
"
|
110 |
-
"
|
111 |
-
"statistical_metrics": [
|
112 |
-
"mean",
|
113 |
-
"std",
|
114 |
-
"median",
|
115 |
-
"ci_lower",
|
116 |
-
"ci_upper",
|
117 |
-
"p_value",
|
118 |
-
"effect_size",
|
119 |
-
],
|
120 |
}
|
121 |
|
122 |
# VALIDATION REQUIREMENTS
|
@@ -124,7 +83,6 @@ VALIDATION_CONFIG = {
|
|
124 |
"min_samples_per_track": {
|
125 |
"google_comparable": 200,
|
126 |
"ug40_complete": 400,
|
127 |
-
"language_pair_matrix": 50,
|
128 |
},
|
129 |
"max_missing_rate": 0.05, # 5% missing predictions allowed
|
130 |
"quality_thresholds": {
|
@@ -135,53 +93,6 @@ VALIDATION_CONFIG = {
|
|
135 |
},
|
136 |
}
|
137 |
|
138 |
-
# UI CONFIGURATION
|
139 |
-
UI_CONFIG = {
|
140 |
-
"title": "🏆 SALT Translation Leaderboard - Scientific Edition",
|
141 |
-
"description": """
|
142 |
-
Rigorous evaluation of translation models on Ugandan languages with statistical significance testing.
|
143 |
-
Three evaluation tracks ensure fair comparison across different model capabilities and language support.
|
144 |
-
""",
|
145 |
-
"tracks": {
|
146 |
-
"google_comparable": {
|
147 |
-
"tab_name": "🤖 Google-Comparable Track",
|
148 |
-
"icon": "🤖",
|
149 |
-
"color": "#1f77b4",
|
150 |
-
},
|
151 |
-
"ug40_complete": {
|
152 |
-
"tab_name": "🌍 UG40-Complete Track",
|
153 |
-
"icon": "🌍",
|
154 |
-
"color": "#ff7f0e",
|
155 |
-
},
|
156 |
-
"language_pair_matrix": {
|
157 |
-
"tab_name": "📊 Language-Pair Matrix",
|
158 |
-
"icon": "📊",
|
159 |
-
"color": "#2ca02c",
|
160 |
-
},
|
161 |
-
},
|
162 |
-
}
|
163 |
-
|
164 |
-
# CHART CONFIGURATION - Research-grade styling
|
165 |
-
CHART_CONFIG = {
|
166 |
-
"statistical_colorscale": "RdYlBu_r",
|
167 |
-
"category_colors": {cat: info["color"] for cat, info in MODEL_CATEGORIES.items()},
|
168 |
-
"heatmap_config": {
|
169 |
-
"colorscale": "Viridis",
|
170 |
-
"show_values": True,
|
171 |
-
"font_size": 10,
|
172 |
-
},
|
173 |
-
"confidence_interval_config": {
|
174 |
-
"alpha": 0.3,
|
175 |
-
"line_width": 2,
|
176 |
-
"marker_size": 8,
|
177 |
-
},
|
178 |
-
"statistical_plot_config": {
|
179 |
-
"height": 600,
|
180 |
-
"width": 800,
|
181 |
-
"margin": {"l": 100, "r": 50, "t": 50, "b": 100},
|
182 |
-
},
|
183 |
-
}
|
184 |
-
|
185 |
# FILE FORMAT SPECIFICATIONS
|
186 |
PREDICTION_FORMAT = {
|
187 |
"required_columns": ["sample_id", "prediction"],
|
@@ -198,8 +109,11 @@ PREDICTION_FORMAT = {
|
|
198 |
# EVALUATION SETTINGS
|
199 |
MAX_TEST_SAMPLES = 500 # Per language pair
|
200 |
MIN_SAMPLES_PER_PAIR = 10 # Minimum for basic statistics
|
201 |
-
|
202 |
-
|
203 |
-
|
204 |
-
"
|
|
|
|
|
|
|
205 |
}
|
|
|
12 |
|
13 |
LANGUAGE_NAMES = {
|
14 |
"ach": "Acholi",
|
15 |
+
"eng": "English",
|
16 |
"lgg": "Lugbara",
|
17 |
"lug": "Luganda",
|
18 |
"nyn": "Runyankole",
|
|
|
24 |
# Google Translate supported subset (for fair comparison)
|
25 |
GOOGLE_SUPPORTED_LANGUAGES = ["lug", "ach", "swa", "eng"]
|
26 |
|
27 |
+
# EVALUATION TRACKS
|
|
|
|
|
|
|
28 |
EVALUATION_TRACKS = {
|
29 |
"google_comparable": {
|
30 |
"name": "Google-Comparable Track",
|
31 |
+
"description": "Models evaluated on language pairs supported by Google Translate for commercial comparison",
|
32 |
"languages": GOOGLE_SUPPORTED_LANGUAGES,
|
33 |
"min_samples_per_pair": 50,
|
|
|
|
|
34 |
},
|
35 |
"ug40_complete": {
|
36 |
+
"name": "UG40-Complete Track",
|
37 |
+
"description": "Models evaluated on all UG40 language pairs for comprehensive assessment",
|
38 |
"languages": ALL_UG40_LANGUAGES,
|
39 |
"min_samples_per_pair": 30,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
},
|
41 |
}
|
42 |
|
|
|
49 |
"color": "#1f77b4",
|
50 |
},
|
51 |
"research": {
|
52 |
+
"name": "Research Models",
|
53 |
"description": "Academic and research institution models",
|
54 |
"examples": ["nllb", "m2m100"],
|
55 |
"color": "#ff7f0e",
|
|
|
62 |
},
|
63 |
"community": {
|
64 |
"name": "Community Submissions",
|
65 |
+
"description": "User-submitted models and fine-tuned variants",
|
66 |
"examples": ["user_submission"],
|
67 |
"color": "#d62728",
|
68 |
},
|
69 |
}
|
70 |
|
71 |
+
# METRICS CONFIGURATION
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
METRICS_CONFIG = {
|
73 |
"primary_metrics": ["bleu", "chrf", "quality_score"],
|
74 |
+
"secondary_metrics": ["rouge1", "rougeL", "cer", "wer"],
|
75 |
"display_precision": 4,
|
76 |
+
"confidence_level": 0.95,
|
77 |
+
"bootstrap_samples": 1000,
|
78 |
+
"min_samples_for_ci": 20,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
79 |
}
|
80 |
|
81 |
# VALIDATION REQUIREMENTS
|
|
|
83 |
"min_samples_per_track": {
|
84 |
"google_comparable": 200,
|
85 |
"ug40_complete": 400,
|
|
|
86 |
},
|
87 |
"max_missing_rate": 0.05, # 5% missing predictions allowed
|
88 |
"quality_thresholds": {
|
|
|
93 |
},
|
94 |
}
|
95 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
96 |
# FILE FORMAT SPECIFICATIONS
|
97 |
PREDICTION_FORMAT = {
|
98 |
"required_columns": ["sample_id", "prediction"],
|
|
|
109 |
# EVALUATION SETTINGS
|
110 |
MAX_TEST_SAMPLES = 500 # Per language pair
|
111 |
MIN_SAMPLES_PER_PAIR = 10 # Minimum for basic statistics
|
112 |
+
|
113 |
+
# CHART CONFIGURATION
|
114 |
+
CHART_CONFIG = {
|
115 |
+
"category_colors": {cat: info["color"] for cat, info in MODEL_CATEGORIES.items()},
|
116 |
+
"height": 600,
|
117 |
+
"width": 800,
|
118 |
+
"margin": {"l": 100, "r": 50, "t": 50, "b": 100},
|
119 |
}
|