Schemes for training and test in ni

Yes does tests