Skip to content

ChanatipSaetia/ESL-NN

Repository files navigation

Encoded Sharing Layers Neural Network (ESL-NN)

Overview

ESL-NN is a "Hierarchical Multi-Label Classification System" based on PyTorch which is free for download and free for non-commecial use.

Features

  • Multi-class classification
  • Multi-label classification
  • Hierarchical multi-label classification with tree and DAG structure.

Supported Platforms

  • Linux
  • MACOS

Team Members

  1. Peerapon Vateekul (Supervisor)
  2. Chanatip Saetia

Folder structure

  • /assemble_classifier - Main classifiers code which store ESLNN, SHLNN, HMCLMLP
  • /classifier - Classifiers of each level code which will be composed together in main classifiers in /assemble_classifier
  • /data - Data preprocessing code and Dataset class which is specifically used for our main classifier
  • /embedding - Document embedding code
  • /evaluation - Evaluation metric code
  • Train.ipynb - A demo jupyter notebook for a training process
  • Test.ipynb - A demo jupyter notebook for a testing process

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published