pkg/train

• train.go