learn vocab over train set only