separate Association class to another file
use CBTrainModel class to maintain train data info