diff = {
let ds_a = datasets["fineweb"]
let ds_b = datasets[dataset]
return features.map((d,i) => {
return {
...d,
dataset_max: ds_b[i].max_activation,
dataset_count: ds_b[i].count,
dataset_avg: ds_b[i].avg_activation,
max_diff: ds_a[i].max_activation - ds_b[i].max_activation,
count_diff: ds_a[i].count/ds_b.length - ds_b[i].count/ds_a.length,
avg_diff: ds_a[i].avg_activation - ds_b[i].avg_activation,
a: ds_a[i],
b: ds_b[i]
}
})
}