Skip to content

This project implements a Bayesian Neural Network (BNN) using PyTorch. A BNN is a type of neural network that treats model weights as probability distributions rather than fixed values. This allows for uncertainty estimation in predictions and robustness against overfitting.

Notifications You must be signed in to change notification settings

bominwang/BayesianNeuralNetwork

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 

Repository files navigation

Bayesian Neural Network (BNN)

Overview

This project implements a Bayesian Neural Network (BNN) using PyTorch. A BNN is a type of neural network that treats model weights as probability distributions rather than fixed values. This allows for uncertainty estimation in predictions and robustness against overfitting.

Installation

Before using this code, make sure you have the required dependencies installed. You can install them using the following command:

pip install -r requirements.txt

About

This project implements a Bayesian Neural Network (BNN) using PyTorch. A BNN is a type of neural network that treats model weights as probability distributions rather than fixed values. This allows for uncertainty estimation in predictions and robustness against overfitting.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages