Added RMSE for Statistics' Regression

minjk-bl · minjk-bl · commit 9fe0c6c0a939 · 2024-05-13T15:46:09.000+09:00
diff --git a/visualpython/html/m_stats/regression.html b/visualpython/html/m_stats/regression.html
@@ -76,6 +76,7 @@
                 <label><input type="checkbox" id="normTest" class="vp-state" checked><span>Residual Normality test</span></label>
                 <label><input type="checkbox" id="histogram" class="vp-state" checked><span>Residual histogram</span></label>
                 <label><input type="checkbox" id="scatterplot" class="vp-state" checked><span>Residual scatterplot</span></label>
+                <label><input type="checkbox" id="rmse" class="vp-state" checked><span>RMSE</span></label>
             </div>
         </div>
     </div>
diff --git a/visualpython/js/m_stats/Regression.js b/visualpython/js/m_stats/Regression.js
@@ -58,6 +58,7 @@ define([
                 normTest: true,
                 histogram: true,
                 scatterplot: true,
+                rmse: false,
                 ...this.state
             };
 
@@ -259,7 +260,7 @@ define([
                 // Multi-collinearity
                 multiCollinearity,
                 // Residual option
-                statistics, normTest, histogram, scatterplot
+                statistics, normTest, histogram, scatterplot, rmse,
             } = this.state;
             let codeList = [];
             let code = new com_String();
@@ -697,7 +698,7 @@ define([
             }
 
             // Residual option
-            if (statistics === true || normTest === true || histogram === true || scatterplot === true) {
+            if (statistics === true || normTest === true || histogram === true || scatterplot === true || rmse === true) {
                 let residualTitle = 'Residual'
                 if (lastModelNum > 0) {
                     residualTitle += ' - Model ' + lastModelNum;
@@ -768,6 +769,14 @@ define([
                     code.appendLine("    plt.tight_layout()");
                     code.append("    plt.show()");
                 }
+                if (rmse === true) {
+                    code.appendLine();
+                    code.appendLine();
+                    code.appendLine("# RMSE (Root Mean Squared Error)");
+                    code.appendLine("_rmse = np.sqrt(_result.mse_resid)");
+                    code.appendLine("display(Markdown('### RMSE (Root Mean Squared Error)'))");
+                    code.append("display(Markdown(f'RMSE: {_rmse}'))");
+                }
             }
 
             codeList.push(code.toString());
