DocumentationNeuronDB Documentation
Documentation Branch: You are viewing documentation for the main branch (3.0.0-devel). Select a branch to view its documentation:

Classification

Classification

NeuronDB supports multiple classification algorithms.

Logistic Regression

CREATE TEMP TABLE logreg_model AS
SELECT neurondb.train(
    'default',
    'logistic_regression',
    'training_table',
    'label',
    ARRAY['features'],
    '{}'::jsonb
)::integer AS model_id;

Prediction

SELECT neurondb.predict(
    (SELECT model_id FROM logreg_model),
    features
) AS prediction
FROM test_table;

Other Algorithms

NeuronDB also supports:

  • SVM (Support Vector Machine)
  • Naive Bayes
  • Decision Trees
  • Neural Networks

Learn More

For detailed documentation on classification algorithms, evaluation metrics, hyperparameter tuning, and multi-class classification, visit:

Classification Documentation