И ещё может быть проблема с тем, что вы пытаетесь очень сильно скомпрессовать признаки. Сколько признаков на входе? Если порядка 100, то это может быть также проблемой