Create dataset for full set of transients in PLAsTiCC dataset
tallamjr opened this issue · 1 comments
tallamjr commented
The functionality to create the dataset for Supernovae only events is already in place. To be able to compare metrics with avocado and others, it would be desirable to test the classifier on the full range of transient events that occur in the PLAsTiCC dataset (all 18 classes).
This would essentially just involve skipping the df = filter_dataframe_only_supernova("../data/plasticc/train_subset.txt", data) function call that reduces the data to only contain Supernovae samples.