After checking the test datasets, i could see that there are two types of datasets namely aoi_2020_02.shp and aoi_2020_01.shp respectively. Is it a must to use these two datasets or we can use only one. The reason for this is that, while i was creating rasterize_vector_data for these two datasets at the same time by converting both to .tif formats, my notebook keeps on crashhing. And these rasterize vector datasets are the ones that i will predict on using my trained model.

So in a knutshell, is it possible to predict on only one rasterize_vector_data or we must predict on the two datasets.

Note: I was able to create rasterize_vector_data from train dataset ( e.g pac_2024_training.shp ) without any issue.