소스 검색

fixed a bug in cv composer

tanja 3 년 전
부모
커밋
7ccd41b6f7
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      cdplib/ml_validation/CVComposer.py

+ 2 - 2
cdplib/ml_validation/CVComposer.py

@@ -76,8 +76,8 @@ class CVComposer:
             X = pd.concat([X_train, X_test], ignore_index=False, axis=0)
             X = pd.concat([X_train, X_test], ignore_index=False, axis=0)
 
 
         else:
         else:
-            cv = self.dummy_cv(train_size=len(X_train),
-                               test_size=len(X_test))
+            cv = self.dummy_cv(train_set_size=len(X_train),
+                               test_set_size=len(X_test))
 
 
             X = np.concatenate([X_train, X_test])
             X = np.concatenate([X_train, X_test])