- Notifications
You must be signed in to change notification settings - Fork328
Open
Description
Expected
We can make predictions using thecatboost model with PGML
Problem
We are running into an error when running predictions when a model is trained as catboost, neither with 'classification' nor 'regression'.
SQL Error [XX000]: ERROR: ValueError: cannot reshape array of size 4 into shape (0)Additional notes
- We were able to make it work using xgboost (classification and regression). Only breaks with catboost
- We also tried passing a different number of arguments (features) to the predict method, but it still fails
Reproduction steps
- Run the "quick start with docker" command (without the GPUs instruction as we don't need them):
docker run \ -it \ -v postgresml_data:/var/lib/postgresql \ -p 5433:5432 \ -p 8000:8000 \ [ghcr.io/postgresml/postgresml:2.10.0](http://ghcr.io/postgresml/postgresml:2.10.0) \ sudo -u postgresml psql -d postgresml
- In another terminal, run the following sql script using psql
CREATE EXTENSION pgml cascade;-- Simple test dataCREATETABLEpgml.training_data ( idSERIALPRIMARY KEY, feature1REAL, feature2REAL, feature3REAL, targetINTEGER);INSERT INTOpgml.training_data (feature1, feature2, feature3, target)VALUES (1.2,3.4,5.6,0), (2.3,4.5,6.7,1), (3.4,5.6,7.8,0), (4.5,6.7,8.9,1), (5.6,7.8,9.0,0);--Training and using a xgboost modelSELECT*FROMpgml.train(project_name=>'classification_xgboost',task=>'classification', relation_name=>'pgml.training_data', y_column_name=>'target', algorithm=>'xgboost');selectpgml.predict( project_name=>'classification_xgboost', features=> ARRAY[10::integer,0.1::real,0.3::real,0.1::real])AS prediction;-- Works well-- Now with catboostSELECT*FROMpgml.train(project_name=>'classification_catboost',task=>'classification', relation_name=>'pgml.training_data', y_column_name=>'target', algorithm=>'catboost');selectpgml.predict( project_name=>'classification_catboost', features=> ARRAY[10::integer,0.1::real,0.3::real,0.1::real])AS prediction;--Throws error:--SQL Error [XX000]: ERROR: ValueError: cannot reshape array of size 4 into shape (0)
Any help regarding this issue would be appreciated. Thank you very much.
Metadata
Metadata
Assignees
Labels
No labels