redo landmark with f1

3springs · Oct 13, 2020 · c2ee155 · c2ee155
1 parent 894a1cc
commit c2ee155
Show file tree

Hide file tree

Showing 3 changed files with 460 additions and 218 deletions.
diff --git a/deep_ml_curriculum/data/landmass_f3.py b/deep_ml_curriculum/data/landmass_f3.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 from torchvision.datasets import MNIST
 
 class LandmassF3Patches(MNIST):
@@ -24,7 +25,7 @@ class LandmassF3Patches(MNIST):
         target_transform (callable, optional): A function/transform that takes in the
             target and transforms it.
     """
-    classes = ['Chaotic Horizon', 'Fault', 'Horizon', 'Salt Dome']
+    classes = ['Discontinuous', 'Faulted', 'Continuous', 'Salt']
 
 
 class LandmassF3PatchesMini(MNIST):
@@ -53,4 +54,6 @@ class LandmassF3PatchesMini(MNIST):
         target_transform (callable, optional): A function/transform that takes in the
             target and transforms it.
     """
-    classes = ['Chaotic Horizon', 'Fault', 'Horizon', 'Salt Dome']
+    classes = ['Discontinuous', 'Faulted', 'Continuous', 'Salt']
+
+