Added correctly values of dataset_collapse and support for learned models (phase I)