package nn // implement train test split function func trainTestSplit() {}