Commit14b1f61

Montana Low

committed

use bytea directly for pl/python rather than hex/text conversion

1 parent958cfba commit14b1f61Copy full SHA for 14b1f61

File tree

-4

lines changed

-4

lines changed

Lines changed: 4 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ INSERT INTO scikit_train_data (value, weight) SELECT generate_series(1, 500), 5.`
`26`	`26`
`27`	`27`
`28`	`28`	`CREATE OR REPLACEFUNCTIONscikit_learn_train_example()`
`29`		`-RETURNSTEXT`
	`29`	`+RETURNSBYTEA`
`30`	`30`	`AS $$`
`31`	`31`	`fromsklearn.ensemble import RandomForestClassifier`
`32`	`32`	`import pickle`
`@@ -45,18 +45,18 @@ AS $$`
`45`	`45`	`rfc= RandomForestClassifier()`
`46`	`46`	`rfc.fit(X, y)`
`47`	`47`
`48`		`- returnpickle.dumps(rfc).hex()`
	`48`	`+ returnpickle.dumps(rfc)`
`49`	`49`
`50`	`50`	`$$ LANGUAGE plpython3u;`
`51`	`51`
`52`	`52`	`;`
`53`	`53`
`54`		`-CREATE OR REPLACEFUNCTIONscikit_learn_predict_example(modelTEXT, valueINT)`
	`54`	`+CREATE OR REPLACEFUNCTIONscikit_learn_predict_example(modelBYTEA, valueINT)`
`55`	`55`	`RETURNSDOUBLE PRECISION`
`56`	`56`	`AS $$`
`57`	`57`	`import pickle`
`58`	`58`
`59`		`- m=pickle.loads(bytes.fromhex(model))`
	`59`	`+ m=pickle.loads(model)`
`60`	`60`
`61`	`61`	`r=m.predict([[value,]])`
`62`	`62`	`return r[0]`

Comments

(0)