search

afeinstein20 / stella

For characterizing flares with convolutional neural networks
https://adina.feinste.in/stella
MIT License
26 stars 18 forks source link

Tutorial: Creating A New CNN - clarification on data format needed #6

Closed ekaterinailin closed 4 years ago

ekaterinailin commented 4 years ago

Hello again,

the Quickstart Tutorial went super smoothly :100: , so I am now at the second part Creating A New CNN. Here I got stuck at 1.1.

As I understood your example on the docs page, I need the path to a folder with light curves, and the path to a .csv file or similar that contains labeled flares.

So what I need to figure out is:

  1. The light curves are supposed to be in .npy format. Should I download a bunch of TESS light cuves (as bulk or using lightkurve), and cast them in .npy with time, flux, and flux error indices?

  2. What should the .csv file look like? Can I use Vizier table1 from Günther+2020?

It would be awesome if you could drop a file or two as an example somewhere in the repo or elsewhere to get an idea. Any guidance in general is appreciated at this point :smiley_cat:

Cheers Ekaterina

afeinstein20 commented 4 years ago

Hi @ekaterinailin Thanks for pointing this out!! I have added a class called "DownloadSets" which will now download the flare catalog from Günther+2020 from Vizier and download/reformat the light curves for the CNN training sets. The tutorial notebook has been updated to reflect this as well :)

Please close the issue if you find it works!

ekaterinailin commented 4 years ago

Awesome, the DownloadSets works as expected, as does the whole tutorial. Thank you!