Commit28545c9

committed

more fixes

1 parent85010b3 commit28545c9Copy full SHA for 28545c9

File tree

-5

lines changed

-5

lines changed

Lines changed: 6 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -208,11 +208,11 @@ def data(self):`
`208`	`208`
`209`	`209`	`# Sanity check the data`
`210`	`210`	`iflen(data)==0:`
`211`		`-PgMLException(`
	`211`	`+raisePgMLException(`
`212`	`212`	f"Relation `{self.relation_name}` contains no rows. Did you pass the correct `relation_name`?"
`213`	`213`	`)`
`214`	`214`	`ifself.y_column_namenotindata[0]:`
`215`		`-PgMLException(`
	`215`	`+raisePgMLException(`
`216`	`216`	f"Column `{self.y_column_name}` not found. Did you pass the correct `y_column_name`?"
`217`	`217`	`)`
`218`	`218`
`@@ -428,6 +428,10 @@ def train(`
`428`	`428`	`algorithms= ["linear","random_forest"]`
`429`	`429`	`elifobjective=="classification":`
`430`	`430`	`algorithms= ["random_forest"]`
	`431`	`+else:`
	`432`	`+raisePgMLException(`
	`433`	`+f"Unknown objective '{objective}', available options are: regression, classification"`
	`434`	`+ )`
`431`	`435`
`432`	`436`	`foralgorithm_nameinalgorithms:`
`433`	`437`	`model=Model.create(project,snapshot,algorithm_name)`

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -103,15 +103,15 @@ $$ LANGUAGE plpython3u;`
`103`	`103`	`---`
`104`	`104`	`--- Regression`
`105`	`105`	`---`
`106`		`-DROPFUNCTIONpgml.train(project_nameTEXT, objectiveTEXT, relation_nameTEXT, y_column_nameTEXT);`
	`106`	`+DROPFUNCTIONIF EXISTSpgml.train(project_nameTEXT, objectiveTEXT, relation_nameTEXT, y_column_nameTEXT);`
`107`	`107`	`CREATE OR REPLACEFUNCTIONpgml.train(project_nameTEXT, objectiveTEXT, relation_nameTEXT, y_column_nameTEXT)`
`108`		`-RETURNSTEXT`
	`108`	`+RETURNSTABLE(project_nameTEXT, objectiveTEXT, statusTEXT)`
`109`	`109`	`AS $$`
`110`	`110`	`frompgml.model import train`
`111`	`111`
`112`	`112`	`train(project_name, objective, relation_name, y_column_name)`
`113`	`113`
`114`		`-return"OK"`
	`114`	`+return[(project_name, objective,"deployed")]`
`115`	`115`	`$$ LANGUAGE plpython3u;`
`116`	`116`
`117`	`117`	`---`

Comments

(0)