refactoring train-test code

訓練とテストだけリファクタリング
あとメソッド名とかそのあたりも