/Semantic-consistent-Embedding

The implementation of "Semantic-consistent Embedding".

Primary LanguagePythonMIT LicenseMIT

Semantic-consistent Embedding

This is the toy data example of semantic-consistent embedding (SCE).

Requirements

python=3.8
pytorch=1.9.0
pandas=1.2.5
scikit-learn=0.24

The code was tested with Pytorch 1.9.0, CUDA 10.2 and Windows 10.

Quick start

Download the dataset from https://www.kaggle.com/esathyaprakash/electrical-fault-detection-and-classification. Move the dataset to [data] folder.

python sce.py

Thanks to the third party

FDAT