SebastianGer / WildfireSpreadTS

MIT License
16 stars 3 forks source link

About number of modalities in input #2

Closed Aamer98 closed 4 months ago

Aamer98 commented 5 months ago

Hey! I was wondering about the difference in the number of modalities (channels) in input mentioned in the paper and documentation, which is 23, and the channel dimension when loading the dataset, which is 40. Did I overlook or miss something here? Thanks!

SebastianGer commented 4 months ago

Hey! I was wondering about the difference in the number of modalities (channels) in input mentioned in the paper and documentation, which is 23, and the channel dimension when loading the dataset, which is 40. Did I overlook or miss something here? Thanks!

Thanks for the question! The difference comes from one-hot-encoding the landcover class. In the raw data it's only an integer, but because it's a nominal variable, it seemed a lot more sensible to use one-hot-encoding when feeding it into the network. Additionally, the active fire is available once as a binary feature, once containing the detection times. If you call FireSpreadDataset.map_channel_index_to_features, you can get the list of all feature names, either with or without the expanded land cover classes, depending on the boolean argument you pass.