work on using training data to get optimized entity set