Commit01c8701

and

authored

Update pgml/pgml/model.py

Co-authored-by: Lev Kokotov <levkk@users.noreply.github.com>

1 parentf61cc0b commit01c8701Copy full SHA for 01c8701

File tree

-1

lines changed

-1

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -214,7 +214,8 @@ def data(self):`
`214`	`214`	`split=self.test_size`
`215`	`215`	`ifisinstance(split,float):`
`216`	`216`	`split=int(self.test_size*X.len())`
`217`		`-returnX[0:split],X[split:X.len()-1],y[0:split],y[split:y.len()-1]`
	`217`	`+returnX[:split],X[split:],y[:split],y[split:]`
	`218`	`+`
`218`	`219`
`219`	`220`	`# TODO normalize and clean data`
`220`	`221`

Comments

(0)