add structure for training counts for observations