Преглед на файлове

fixed a bug in cv composer

tanja преди 3 години
родител
ревизия
4f70ffd515
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      cdplib/ml_validation/CVComposer.py

+ 2 - 2
cdplib/ml_validation/CVComposer.py

@@ -70,8 +70,8 @@ class CVComposer:
 
         if use_index:
 
-            cv = self.dummy_cv(train_index=X_train.index,
-                               test_index=X_test.index)
+            cv = self.dummy_cv(train_set_index=X_train.index,
+                               test_set_index=X_test.index)
 
             X = pd.concat([X_train, X_test], ignore_index=False, axis=0)