remove stuff from cross validation

floriangriese · Nov 23, 2022 · 7372637 · 7372637
1 parent fb90c12
commit 7372637
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 30 deletions.
diff --git a/README.md b/README.md
@@ -43,15 +43,6 @@ The first splitting option (galaxy_data_h5.zip) provides three splittings in tra
 | test   | 50        | 50       | 50       | 50      |200       |
 | total   | 495        |924       |391       |348       |2158       |
 
-The second splitting option (galaxy_data_crossvalid_0_h5.zip to galaxy_data_crossvalid_4_h5.zip and galaxy_data_crossvalid_test_h5.zip) provides a 5-fold cross validation dataset with a larger test set.
-
-| classes/split     | FRI |   FRII |     Compact |    Bent |   Total     |
-| ----------- | ----------- |----------- |----------- |-----------       |-----------|
-| 5-fold cross train   | 316        | 659       | 232       | 198      |1405       |
-| 5-fold cross valid   | 79        | 165       | 59       | 50      |353       |
-| test   | 100        | 100       | 100       | 100      |400       |
-| total   | 495        |924       |391       |348       |2158       |
-
 # Installation usage with pytorch
 If you want to use the dataset via the dataset class `FIRSTGalaxyData` with pytorch, install the necessary packages with
 

diff --git a/firstgalaxydata/Example_firstgalaxydata.py b/firstgalaxydata/Example_firstgalaxydata.py
@@ -27,24 +27,5 @@
                            is_PIL=True, is_RGB=True, transform=transformRGB)
     print(data)
 
-    # Usage of 5-fold cross validation set
-
-    data = FIRSTGalaxyData(root="./", selected_split="train", input_data_list=["galaxy_data_crossvalid_0_h5.h5"],
-                           is_PIL=True, is_RGB=True, transform=transformRGB)
-
-    print(data)
-
-    data = FIRSTGalaxyData(root="./", selected_split="valid", input_data_list=["galaxy_data_crossvalid_0_h5.h5"],
-                           is_PIL=True, is_RGB=True, transform=transformRGB)
-
-    print(data)
-
-    data = FIRSTGalaxyData(root="./", selected_split="test", input_data_list=["galaxy_data_crossvalid_test_h5.h5"],
-                           selected_classes=["FRI", "FRII", "Compact", "Bent"],
-                           selected_catalogues=["Gendre", "MiraBest", "Capetti2017a", "Capetti2017b",
-                                                "Baldi2018", "Proctor_Tab1"],
-                           is_PIL=True, is_RGB=True, transform=transformRGB)
-    print(data)
-
 
 
diff --git a/firstgalaxydata/firstgalaxydata.py b/firstgalaxydata/firstgalaxydata.py
@@ -9,8 +9,6 @@
 from astropy.coordinates import SkyCoord
 from torchvision.datasets.utils import download_url
 import zipfile
-#from importlib import resources
-#import io
 
 
 class FIRSTGalaxyData(data.Dataset):