joelniklaus HF Staff commited on
Commit
cb15587
·
1 Parent(s): a45cf8d

standardize model family colors across plots

Browse files
app/src/content/chapters/experiments.mdx CHANGED
@@ -5,9 +5,8 @@ import Glossary from "../../components/Glossary.astro";
5
  import FigRef from "../../components/FigRef.astro";
6
 
7
  {/* TODO: mention the currently running finephrase rephrasing with smollm2 */}
8
- {/* TODO: shorten the vllm inference benchmark or put stuff into the appendix */}
9
  {/* TODO: potentially make a widget for data exploration: look at the same few samples generated by different models or transformed with different prompts */}
10
- {/* TODO: Standardize colors across charts in the blog post more */}
11
  {/* TODO: Integrate decay experiment as another analysis for proxy */}
12
 
13
  {/*
 
5
  import FigRef from "../../components/FigRef.astro";
6
 
7
  {/* TODO: mention the currently running finephrase rephrasing with smollm2 */}
8
+ {/* TODO: read through entire blog post and make improvements */}
9
  {/* TODO: potentially make a widget for data exploration: look at the same few samples generated by different models or transformed with different prompts */}
 
10
  {/* TODO: Integrate decay experiment as another analysis for proxy */}
11
 
12
  {/*
app/src/content/embeds/d3-optimization-sweep.html CHANGED
@@ -168,17 +168,9 @@
168
  const margin = { top: 20, right: 40, bottom: 40, left: 130 };
169
 
170
  // ── Colors & shapes ──
171
- function getFamilyColors() {
172
- try {
173
- if (window.ColorPalettes && typeof window.ColorPalettes.getColors === 'function') {
174
- return window.ColorPalettes.getColors('categorical', FAMILIES.length);
175
- }
176
- } catch (e) {}
177
- return ['var(--primary-color)', '#e377c2', '#17becf', '#bcbd22'];
178
- }
179
-
180
- const familyPalette = getFamilyColors();
181
- const familyColor = (family) => familyPalette[FAMILIES.indexOf(family)] || 'var(--primary-color)';
182
 
183
  const shapeGenerators = {
184
  'Baseline': d3.symbol().type(d3.symbolCircle),
 
168
  const margin = { top: 20, right: 40, bottom: 40, left: 130 };
169
 
170
  // ── Colors & shapes ──
171
+ const FAMILY_COLORS = { 'Qwen3': '#e07b54', 'SmolLM2': '#e06b9e', 'Gemma3': '#5b9bd5', 'GPT-OSS': '#8bc474' };
172
+ const familyPalette = FAMILIES.map(f => FAMILY_COLORS[f] || '#999');
173
+ const familyColor = (family) => FAMILY_COLORS[family] || '#999';
 
 
 
 
 
 
 
 
174
 
175
  const shapeGenerators = {
176
  'Baseline': d3.symbol().type(d3.symbolCircle),
app/src/content/embeds/experiment-overview.html CHANGED
@@ -142,8 +142,8 @@
142
 
143
  // Colors
144
  const sourceColors = { 'FW-Edu HQ': '#6B8DB5', 'FW-Edu LQ': '#B58B9B', 'DCLM': '#7B82C8', 'Cosmopedia': '#8BA878' };
145
- const catColors = { 'Format': '#4EA5B7', 'Nemotron': '#D4A850', 'REWIRE': '#C87878' };
146
- const familyColors = { 'Gemma': '#4EA5B7', 'Qwen': '#8B7BE8', 'SmolLM2': '#E8C44A', 'Falcon': '#E889AB', 'Granite': '#5BC0A4', 'Llama': '#D09090' };
147
 
148
  const nodeColor = (d) => {
149
  if (d.col === 'source') return sourceColors[d.name] || '#888';
 
142
 
143
  // Colors
144
  const sourceColors = { 'FW-Edu HQ': '#6B8DB5', 'FW-Edu LQ': '#B58B9B', 'DCLM': '#7B82C8', 'Cosmopedia': '#8BA878' };
145
+ const catColors = { 'Format': '#4EA5B7', 'Nemotron': '#76b900', 'REWIRE': '#1877F2' };
146
+ const familyColors = { 'Gemma': '#5b9bd5', 'Qwen': '#e07b54', 'SmolLM2': '#e06b9e', 'Falcon': '#c9a046', 'Granite': '#9a8ec2', 'Llama': '#8bc474' };
147
 
148
  const nodeColor = (d) => {
149
  if (d.col === 'source') return sourceColors[d.name] || '#888';