Face and Digit Classification

A implementation of Naive Bayes algorithm and perceptron algorithm
Explore the docs »

View Demo · Report Bug · Request Feature

Table of Contents

1. About The Project
   • Built with
   • Dataset
   • Test result
2. Usage
3. License
4. Contact

About The Project

This is a course project of Rutgers CS440 Intro to Artificial Intelligence.

Built with

• Language:

Dataset

The data folder contains two different types of images: digits from 0-9 and human faces.

Each type of data is splited to three parts:

• Training dataset
• Validation dataset
• Test dataset

Size of a digit image is 28 x 28.

Size of a face image is 60 x 70.

Test result

Average training time

• Digit

  

• Face

  

Accuracy and standard deviation

Digit using Naive Bayes

Digit using Perceptron

Face using Naive Bayes

Face using Perceptron

Weight visualization examples

• Digit

  

• Face

  

Usage

Classify face and digit by using naive bayes algorithm and perceptron algorithm.

License

Distributed under the MIT License.

See LICENSE for more information.

Contact

Feiyu Zheng - [email protected]

Project Link: https://github.com/ChaserZ98/Face-and-Digit-Classification

(back to top)