add trainbase class

add trainbase class