Bug fix

fixed a bug in constructing the training sub feature set