Skip to content

Transfer learning for binary classification on chemical gas sensor data

Notifications You must be signed in to change notification settings

lav30/Transfer_Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 
 
 

Repository files navigation

Alt text

Alt text

Table of Contents

Portfolio Link

Portfolio that elaborates the methodologies used in this project to optimize the chosen metrics for classification.

Project Description

An important assumption in machine learning frameworks and algorithms is that the training and test sets must have similar label distributions and that a dataset must have thousands of observations to produce a good model. Oftentimes, real-world datasets have dissimilar label distributions and few observations to generate an effective model performance. In such cases, model performance can be improved, if the knowledge acquired through a network trained on a large dataset can be transferred successfully to smaller datasets. This project focuses on the proof of concept of transfer learning between gas sensor datasets. The experimental results indicate significant improvements in model test accuracy and other performance metrics upon applying transfer learning.

GitHub last commit GitHub All Releases

Project Poster

Alt text

Installation

(Back to top)

  1. Install Anaconda and all necessary libraries ( pandas, scikit-learn, matplotlib, numpy, etc)
  2. Install Tensorflow

Dataset Information

(Back to top)

Three chemical gas sensor datasets have been used in this project to build models utilised for transfer learning. A large dataset with millions of observations serves as the source dataset and two, much smaller datasets serve as the target datasets. The aim is to use pre-trained networks as initializations for datasets with fewer observations.

  1. Source Dataset
  2. Target Dataset 1
  3. Target Dataset 2

License

(Back to top)

GNU General Public License, version 3 (GPLv3)