https://github.com/salamanderxing/jax-gaussian-bayes
A Multivariate Gaussian Bayes classifier written using JAX
https://github.com/salamanderxing/jax-gaussian-bayes
cifar10 gaussian-bayesian-processes gaussian-naive-bayes jax multivariate-gaussian naive-bayes-classifier numpy xla
Last synced: 3 months ago
JSON representation
A Multivariate Gaussian Bayes classifier written using JAX
- Host: GitHub
- URL: https://github.com/salamanderxing/jax-gaussian-bayes
- Owner: SalamanderXing
- License: mit
- Created: 2022-09-22T14:10:56.000Z (almost 4 years ago)
- Default Branch: main
- Last Pushed: 2022-09-22T14:20:58.000Z (almost 4 years ago)
- Last Synced: 2025-10-12T03:28:27.111Z (9 months ago)
- Topics: cifar10, gaussian-bayesian-processes, gaussian-naive-bayes, jax, multivariate-gaussian, naive-bayes-classifier, numpy, xla
- Language: Python
- Homepage:
- Size: 5.86 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Multivariate Gaussian Bayes classsifier in JAX
This is a simple implementation of a multivariate Gaussian Bayes classifier in [JAX](https://github.com/google/jax).
The classifier is trained and tested on the [CIFAR-10 dataset](https://www.cs.toronto.edu/~kriz/cifar.html).
## Run by:
```
python evaluate.py
```
It will print the accuracy of a random classifier vs a Navie Bayes classifier vs a Multivariate Gaussian Bayes classifier.
If the program fais, probably you need to download the dataset first. You can do it by running:
```
wget https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
```
## Dependencies
- python >= 3.9
- [JAX](https://github.com/google/jax)