def de_c2(X,y): dim = X.shape[1] de = min(2000,dim) clf = SelectKBest(chi2, k = de) clf.fit(X,y) def _func(X1,X2): return clf.transform(X1),clf.transform(X2) return _func