Clean up some tests

Author: Irina Nicolae

art/classifiers/pytorch_unittest.py

@@ -170,7 +170,7 @@ def test_layers(self):
 #         return out
 
 
-class TestTFTextClassifier(unittest.TestCase):
+class TestPyTorchTextClassifier(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         # Load IMDB
@@ -245,7 +245,6 @@ def test_loss_gradient(self):
     def test_layers(self):
         # Get IMDB
         (_, _), (x_test, y_test) = self.imdb
-        y_test = to_categorical(y_test, nb_classes=2)
 
         # Test and get layers
         layer_names = self.classifier.layer_names
@@ -267,8 +266,7 @@ def test_layers(self):
 
     def test_embedding(self):
         # Get IMDB
-        (x_train, y_train), (x_test, y_test) = self.imdb
-        y_train = to_categorical(y_train, nb_classes=2)
+        (_, _), (x_test, y_test) = self.imdb
 
         # Test to embedding
         x_emb = self.classifier.to_embedding(x_test)
@@ -287,10 +285,3 @@ def test_embedding(self):
 
 if __name__ == '__main__':
     unittest.main()
-
-
-
-
-
-
-