import numpy as np
META={'name':'simplicial_node_classification','domain':'graph-nn','description':'Node classification on a two-community cyclic triangle simplicial complex.'}
def get_dataset(seed,n_train,n_test):
    rng=np.random.RandomState(seed); n,f=24,8
    def make(m,rs):
        y=np.zeros((m,n),dtype=np.int64); y[:,n//2:]=1
        x=rs.normal(0,1,(m,n,f)).astype(np.float32); x[:,:,0]+=(2*y-1)*.45
        return x,y
    xtr,ytr=make(n_train,rng); xte,yte=make(n_test,np.random.RandomState(seed+5000))
    return {'xtr':xtr,'ytr':ytr,'xte':xte,'yte':yte,'task':'classification','metric':'err','out_dim':2}