rymc / n2d

A deep clustering algorithm. Code to reproduce results for our paper N2D: (Not Too) Deep Clustering via Clustering the Local Manifold of an Autoencoded Embedding.
GNU General Public License v3.0
129 stars 21 forks source link

how to use n2d in other datasets? #21

Open db-beep opened 2 years ago

db-beep commented 2 years ago

Hi,author, I want to apply my own data set to n2d.data.shape[90,20000],sample:90,feature:20000,what should I do? thanks

rymc commented 2 years ago

Hi

Please have a look at this file: https://github.com/rymc/n2d/blob/master/datasets.py

You can create a new function that loads and returns your data by doing something similar to the functions in datasets.py

def load_har() might be the most useful, since it loads a CSV from disk.