garethjns / Kaggle-EEG

Seizure prediction from EEG data using machine learning. 3rd place solution for Kaggle/Uni Melbourne seizure prediction competition.
101 stars 29 forks source link

Generating figures for Feature comparison #19

Closed YasminAMassoud closed 3 years ago

YasminAMassoud commented 3 years ago

image

How are These plotted ?

Thanks

garethjns commented 3 years ago

Hi @YasminAMassoud, that figure was generated manually, I don't think the code is anywhere in the repo. It shows the mean importance (either gain or number of splits, I forget) for each feature in the random forest model. The top figure is for all the features, unlabelled but grouped by window size. The bottom figure are the top 20 features, with labels.