"""Кастомный трек «token_expert_sequence» — добавлен экспериментом.
Провенанс: эксперимент #714 «Maximum-cardinality token-expert routing»
Дата: 2026-09-01 01:44
Автоматическая приёмка: смок-тест пройден."""
import numpy as np
META={"name":"token_expert_sequence","domain":"moe-routing","description":"Synthetic multi-token classification where each example contains correlated tokens processed by capacity-limited experts."}
def get_dataset(seed,n_train,n_test):
    def gen(s,n):
        rng=np.random.default_rng(s); y=rng.integers(0,2,n)
        # class controls a shared latent direction; tokens are correlated observations
        z=(2*y-1)[:,None,None].astype(np.float32)
        x=z*np.array([1.,-.7,.45],np.float32)[None,None,:]+rng.normal(0,.85,(n,8,3)).astype(np.float32)
        x += rng.normal(0,.25,(n,1,3)).astype(np.float32)
        return x,y.astype(np.int64)
    xtr,ytr=gen(seed,n_train); xte,yte=gen(seed+5000,n_test)
    return {"xtr":xtr,"ytr":ytr,"xte":xte,"yte":yte,"task":"classification","metric":"err","out_dim":2}