ML > data split

minjk-bl · minjk-bl · commit fa2891634905 · 2022-03-02T11:34:38.000+09:00
diff --git a/css/popupComponent.css b/css/popupComponent.css
@@ -283,6 +283,11 @@
 .vp-popup-frame input[type=number]::placeholder {
 	color: var(--gray-color);
 }
+.vp-popup-frame input[type=text]:disabled,
+.vp-popup-frame input[type=number]:disabled {
+    background: var(--light-gray-color);
+    cursor: not-allowed;
+}
 
 .vp-popup-frame select {
     width: 160px;
@@ -292,7 +297,7 @@
 	font-size: 14px;
 	line-height: 16px;
 	padding: 5px 16px 3px 9px;
-	background: url(../img/unfold_more.svg), #F5F5F5;
+	background: url(../img/unfold_more.svg), var(--light-gray-color);
 	background-position: 97% 50%;
 	background-size: 11px 11px;
 	background-repeat: no-repeat;
diff --git a/css/root.css b/css/root.css
@@ -439,6 +439,9 @@ hr.vp-extra-menu-line {
     right: 25px;
     cursor: pointer;
 }
+.vp-vs-box input:disabled + .vp-vs-filter {
+    cursor: not-allowed;
+}
 .vp-vs-filter-box {
     display: none;
     position: absolute;
diff --git a/js/com/component/VarSelector2.js b/js/com/component/VarSelector2.js
@@ -51,8 +51,11 @@ define([
             let that = this;
 
             $(this._parentTag).on('click', this.wrapSelector('.vp-vs-filter'), function() {
-                // toggle filter box
-                $(that.wrapSelector('.vp-vs-filter-box')).toggle();
+                // check disabled
+                if (!$(this).parent().find('input').is(':disabled')) {
+                    // toggle filter box
+                    $(that.wrapSelector('.vp-vs-filter-box')).toggle();
+                }
             });
         }
 
diff --git a/js/m_ml/dataSplit.js b/js/m_ml/dataSplit.js
@@ -36,6 +36,7 @@ define([
                 featureData: '',
                 targetData: '',
                 testSize: 0.25,
+                shuffle: 'True',
                 trainFeatures: 'X_train',
                 trainTarget: 'y_train',
                 testFeatures: 'X_test',
@@ -74,6 +75,18 @@ define([
                     $(that.wrapSelector('#testFeatures')).val('test').trigger('change');
                 }
             });
+
+            // Stratify depends on Shuffle
+            $(this.wrapSelector('#shuffle')).on('change', function() {
+                let shuffle = $(this).val();
+                if (shuffle == 'True') {
+                    // enable stratify
+                    $(that.wrapSelector('#stratify')).prop('disabled', false);
+                } else {
+                    // disable stratify
+                    $(that.wrapSelector('#stratify')).prop('disabled', true);
+                }
+            });
         }
 
         templateForBody() {
@@ -82,7 +95,7 @@ define([
             // test size generating
             let sizeOptions = '';
             for (let i=5; i<95; i+=5) {
-                sizeOptions += `<option value="0.${i}" ${this.state.testSize==('0.'+i)?'selected':''}>
+                sizeOptions += `<option value="${i / 100}" ${parseFloat(this.state.testSize)==(i / 100)?'selected':''}>
                     ${i}%${i==25?' (default)':''}
                 </option>`;
             }
@@ -106,7 +119,7 @@ define([
             varSelector = new VarSelector2(this.wrapSelector(), ['DataFrame', 'List', 'string']);
             varSelector.setComponentID('stratify');
             varSelector.addClass('vp-state vp-input');
-            varSelector.setValue(this.state.targetData);
+            varSelector.setValue(this.state.stratify);
             varSelector.setPlaceholder('None');
             $(page).find('#stratify').replaceWith(varSelector.toTagString());
             
@@ -152,7 +165,7 @@ define([
             let { 
                 trainFeatures, trainTarget, testFeatures, testTarget,
                 inputData, featureData, targetData,
-                testSize, randomState, shuffle, startify
+                testSize, randomState, shuffle, stratify
             } = this.state;
 
             let options = new com_String();
@@ -162,11 +175,11 @@ define([
             if (randomState && randomState != '') {
                 options.appendFormat(', random_state={0}', randomState);
             }
-            if (shuffle && shuffle != '') {
+            if (shuffle && shuffle != 'True') {
                 options.appendFormat(', shuffle={0}', shuffle);
             }
-            if (startify && startify != '') {
-                options.appendFormat(', startify={0}', startify);
+            if (shuffle != 'False' && stratify && stratify != '') {
+                options.appendFormat(', startify={0}', stratify);
             }
 
             let code = new com_String();
