From 540db83bcf28b63d6e3f72c41f69c6c7777f5630 Mon Sep 17 00:00:00 2001
From: minjk-bl <mjkim-bl@gmail.com>
Date: Mon, 14 Mar 2022 19:08:10 +0900
Subject: [PATCH] Fix datasets bug - linnerud has multiple targets

---
 js/m_ml/DataSets.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/js/m_ml/DataSets.js b/js/m_ml/DataSets.js
index 479bdf02..9ad3fe79 100644
--- a/js/m_ml/DataSets.js
+++ b/js/m_ml/DataSets.js
@@ -138,7 +138,11 @@ define([
                 code.appendFormatLine('{0} = {1}', allocateTo, modelCode);
                 code.appendLine("# Create DataFrame");
                 code.appendFormatLine("df_{0} = pd.DataFrame(data={1}.data, columns={2}.feature_names)", allocateTo, allocateTo, allocateTo);
-                code.appendFormat("df_{0}['target'] = {1}.target", allocateTo, allocateTo);
+                if (loadType == 'load_linnerud') {
+                    code.appendFormat("df_{0}[{1}.target_names] = {2}.target", allocateTo, allocateTo, allocateTo);
+                } else {
+                    code.appendFormat("df_{0}['target'] = {1}.target", allocateTo, allocateTo);
+                }
                 allocateToVar = 'df_' + allocateTo;
             } else {
                 code.appendFormatLine("_X, _y = {0}", modelCode);