Aktualisieren structures.py

structures.py

@@ -88,6 +88,8 @@ class PreprocessedDataSet:
                 self.length_list[i] = self.length_list[i-1] + self.length_list[i] 
 
         elif dataset != None and model_gensim != None and nlp != None:
+
+            self.path = path
         
             self.path.mkdir()
                 
@@ -128,12 +130,14 @@ class PreprocessedDataSet:
         
         return dp
         
-    def compute_sent_vecs(self, model):
-    
+    def compute_sent_vecs(self, model, path):
+        
+        path = pathlib.Path(path)
+        path.mkdir()
         for i in range(self.length):
             dp = self.__getitem__(i)
             dp.compute_sent_vecs(model)
-            torch.save(dp, self.path/str(i))
+            torch.save(dp, path/str(i))
     
     # Shorten dataset for testing
     def first(self, k):