From 4c22a2b41f656013ffadf22123e155d69c6eba6a Mon Sep 17 00:00:00 2001 From: Montana Low Date: Wed, 15 Mar 2023 11:29:22 -0700 Subject: [PATCH] disambiguate api --- .../examples/image_classification.sql | 4 +-- pgml-extension/src/api.rs | 27 ++++++++++++++++++- 2 files changed, 28 insertions(+), 3 deletions(-) diff --git a/pgml-extension/examples/image_classification.sql b/pgml-extension/examples/image_classification.sql index 33e6fd773..278195f89 100644 --- a/pgml-extension/examples/image_classification.sql +++ b/pgml-extension/examples/image_classification.sql @@ -22,7 +22,7 @@ SELECT left(image::text, 40) || ',...}', target FROM pgml.digits LIMIT 10; SELECT * FROM pgml.train('Handwritten Digits', 'classification', 'pgml.digits', 'target'); -- check out the predictions -SELECT target, pgml.predict('Handwritten Digits', image::FLOAT4[]) AS prediction +SELECT target, pgml.predict('Handwritten Digits', image) AS prediction FROM pgml.digits LIMIT 10; @@ -95,6 +95,6 @@ SELECT * FROM pgml.deploy('Handwritten Digits', 'rollback'); SELECT * FROM pgml.deploy('Handwritten Digits', 'best_score', 'svm'); -- check out the improved predictions -SELECT target, pgml.predict('Handwritten Digits', image::FLOAT4[]) AS prediction +SELECT target, pgml.predict('Handwritten Digits', image) AS prediction FROM pgml.digits LIMIT 10; diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs index cf966e21a..20f3cda7f 100644 --- a/pgml-extension/src/api.rs +++ b/pgml-extension/src/api.rs @@ -414,10 +414,35 @@ fn deploy( } #[pg_extern(strict, name = "predict")] -fn predict(project_name: &str, features: Vec) -> f32 { +fn predict_f32(project_name: &str, features: Vec) -> f32 { predict_model(Project::get_deployed_model_id(project_name), features) } +#[pg_extern(strict, name = "predict")] +fn predict_f64(project_name: &str, features: Vec) -> f32 { + predict_f32(project_name, features.iter().map(|&i| i as f32).collect()) +} + +#[pg_extern(strict, name = "predict")] +fn predict_i16(project_name: &str, features: Vec) -> f32 { + predict_f32(project_name, features.iter().map(|&i| i as f32).collect()) +} + +#[pg_extern(strict, name = "predict")] +fn predict_i32(project_name: &str, features: Vec) -> f32 { + predict_f32(project_name, features.iter().map(|&i| i as f32).collect()) +} + +#[pg_extern(strict, name = "predict")] +fn predict_i64(project_name: &str, features: Vec) -> f32 { + predict_f32(project_name, features.iter().map(|&i| i as f32).collect()) +} + +#[pg_extern(strict, name = "predict")] +fn predict_bool(project_name: &str, features: Vec) -> f32 { + predict_f32(project_name, features.iter().map(|&i| i as u8 as f32).collect()) +} + #[pg_extern(strict, name = "predict_proba")] fn predict_proba(project_name: &str, features: Vec) -> Vec { predict_model_proba(Project::get_deployed_model_id(project_name), features) pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

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:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy