add label encoding

floriangriese · Sep 26, 2022 · 9135455 · 9135455
1 parent 3c6f2f1
commit 9135455
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -24,6 +24,13 @@ the dataset contains the following number of samples per class.
 | test   | 50        | 50       | 50       | 50      |200       |
 | total   | 495        |924       |391       |348       |2158       |
 
+| classes    | Label |  
+| ----------- | ----------- |
+| FRI    | 0  |
+| FRII   | 1  | 
+| Compact| 2  | 
+| Bent   | 3  |
+
 # Installation usage with pytorch
 If you want to use the dataset via the dataset class `FIRSTGalaxyData` with pytorch, install the necessary packages with