Fixed MLP training data counting