Spark training for entity topic model