Commit540db83

minjk-bl

committed

Fix datasets bug - linnerud has multiple targets

1 parentd6e3c8c commit540db83Copy full SHA for 540db83

File tree

-1

lines changed

-1

lines changed

Lines changed: 5 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,11 @@ define([`
`138`	`138`	`code.appendFormatLine('{0} = {1}',allocateTo,modelCode);`
`139`	`139`	`code.appendLine("# Create DataFrame");`
`140`	`140`	`code.appendFormatLine("df_{0} = pd.DataFrame(data={1}.data, columns={2}.feature_names)",allocateTo,allocateTo,allocateTo);`
`141`		`-code.appendFormat("df_{0}['target'] = {1}.target",allocateTo,allocateTo);`
	`141`	`+if(loadType=='load_linnerud'){`
	`142`	`+code.appendFormat("df_{0}[{1}.target_names] = {2}.target",allocateTo,allocateTo,allocateTo);`
	`143`	`+}else{`
	`144`	`+code.appendFormat("df_{0}['target'] = {1}.target",allocateTo,allocateTo);`
	`145`	`+}`
`142`	`146`	`allocateToVar='df_'+allocateTo;`
`143`	`147`	`}else{`
`144`	`148`	`code.appendFormatLine("_X, _y = {0}",modelCode);`

Comments

(0)