diff --git a/pgml-extension/build.rs b/pgml-extension/build.rs deleted file mode 100644 index e7db2339e..000000000 --- a/pgml-extension/build.rs +++ /dev/null @@ -1,3 +0,0 @@ -fn main() { - println!("cargo:rustc-link-lib=static=openblas"); -} diff --git a/pgml-extension/examples/image_classification.sql b/pgml-extension/examples/image_classification.sql index 278195f89..d1781d285 100644 --- a/pgml-extension/examples/image_classification.sql +++ b/pgml-extension/examples/image_classification.sql @@ -65,6 +65,12 @@ SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'xgboost', hyperpara -- Histogram Gradient Boosting is too expensive for normal tests on even a toy dataset -- SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'hist_gradient_boosting', hyperparams => '{"max_iter": 2}'); +-- runtimes +SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'linear', runtime => 'python'); +SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'linear', runtime => 'rust'); + +SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'xgboost', runtime => 'python', hyperparams => '{"n_estimators": 10}'); +SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'xgboost', runtime => 'rust', hyperparams => '{"n_estimators": 10}'); -- check out all that hard work SELECT trained_models.* FROM pgml.trained_models diff --git a/pgml-extension/examples/regression.sql b/pgml-extension/examples/regression.sql index 6706e2391..000afd2b5 100644 --- a/pgml-extension/examples/regression.sql +++ b/pgml-extension/examples/regression.sql @@ -85,6 +85,12 @@ SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'xgboost', hyperpa -- Histogram Gradient Boosting is too expensive for normal tests on even a toy dataset -- SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'hist_gradient_boosting', hyperparams => '{"max_iter": 10}'); +-- runtimes +SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'linear', runtime => 'python'); +SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'linear', runtime => 'rust'); + +SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'xgboost', runtime => 'python', hyperparams => '{"n_estimators": 10}'); +SELECT * FROM pgml.train('Diabetes Progression', algorithm => 'xgboost', runtime => 'rust', hyperparams => '{"n_estimators": 10}'); -- check out all that hard work SELECT trained_models.* FROM pgml.trained_models diff --git a/pgml-extension/examples/transformers.sql b/pgml-extension/examples/transformers.sql index e7fabbb7d..16b274cd5 100644 --- a/pgml-extension/examples/transformers.sql +++ b/pgml-extension/examples/transformers.sql @@ -2,6 +2,8 @@ \set ON_ERROR_STOP true \timing on +SELECT pgml.embed('intfloat/e5-small', 'hi mom'); + SELECT pgml.transform( 'translation_en_to_fr', @@ -88,4 +90,3 @@ SELECT pgml.transform( ] ) AS answer; - diff --git a/pgml-extension/tests/test.sql b/pgml-extension/tests/test.sql index ece0cd165..1c4dd614b 100644 --- a/pgml-extension/tests/test.sql +++ b/pgml-extension/tests/test.sql @@ -27,4 +27,5 @@ SELECT pgml.load_dataset('wine'); \i examples/multi_classification.sql \i examples/regression.sql \i examples/vectors.sql - +-- transformers are generally too slow to run in the test suite +--\i examples/transformers.sql
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies: