This is the toy data example of semantic-consistent embedding (SCE).
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.
Download the dataset from https://www.kaggle.com/esathyaprakash/electrical-fault-detection-and-classification. Move the dataset to [data] folder.
python sce.py