GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → vlukiyanov → Pt Dec

vlukiyanov / Pt Dec

Licence: mit
PyTorch implementation of DEC (Deep Embedding Clustering)

Programming Languages

python
139335 projects - #7 most used programming language

Labels

pytorchclusteringautoencoder

Projects that are alternatives of or similar to Pt Dec

DESOM
🌐 Deep Embedded Self-Organizing Map: Joint Representation Learning and Self-Organization
Stars: ✭ 76 (-42.42%)
Mutual labels:  clustering, autoencoder
Danmf
A sparsity aware implementation of "Deep Autoencoder-like Nonnegative Matrix Factorization for Community Detection" (CIKM 2018).
Stars: ✭ 161 (+21.97%)
Mutual labels:  autoencoder, clustering
Tensorflow Book
Accompanying source code for Machine Learning with TensorFlow. Refer to the book for step-by-step explanations.
Stars: ✭ 4,448 (+3269.7%)
Mutual labels:  autoencoder, clustering
Timeseries Clustering Vae
Variational Recurrent Autoencoder for timeseries clustering in pytorch
Stars: ✭ 190 (+43.94%)
Mutual labels:  autoencoder, clustering
Pt Sdae
PyTorch implementation of SDAE (Stacked Denoising AutoEncoder)
Stars: ✭ 72 (-45.45%)
Mutual labels:  autoencoder, clustering
Dat8
General Assembly's 2015 Data Science course in Washington, DC
Stars: ✭ 1,516 (+1048.48%)
Mutual labels:  clustering
Hazelcast Nodejs Client
Hazelcast IMDG Node.js Client
Stars: ✭ 124 (-6.06%)
Mutual labels:  clustering
Lstm Autoencoders
Anomaly detection for streaming data using autoencoders
Stars: ✭ 113 (-14.39%)
Mutual labels:  autoencoder
Pytorch cpp
Deep Learning sample programs using PyTorch in C++
Stars: ✭ 114 (-13.64%)
Mutual labels:  autoencoder
Libcluster
An extensible C++ library of Hierarchical Bayesian clustering algorithms, such as Bayesian Gaussian mixture models, variational Dirichlet processes, Gaussian latent Dirichlet allocation and more.
Stars: ✭ 129 (-2.27%)
Mutual labels:  clustering
Tybalt
Training and evaluating a variational autoencoder for pan-cancer gene expression data
Stars: ✭ 126 (-4.55%)
Mutual labels:  autoencoder
Imagecluster
Cluster images based on image content using a pre-trained deep neural network, optional time distance scaling and hierarchical clustering.
Stars: ✭ 122 (-7.58%)
Mutual labels:  clustering
Ml Email Clustering
Email clustering with machine learning
Stars: ✭ 116 (-12.12%)
Mutual labels:  clustering
Adaptive clustering
Lightweight and Accurate Point Cloud Clustering
Stars: ✭ 125 (-5.3%)
Mutual labels:  clustering
Clust4j
A suite of classification clustering algorithm implementations for Java. A number of partitional, hierarchical and density-based algorithms including DBSCAN, k-Means, k-Medoids, MeanShift, Affinity Propagation, HDBSCAN and more.
Stars: ✭ 115 (-12.88%)
Mutual labels:  clustering
Tiny ml
numpy 实现的 周志华《机器学习》书中的算法及其他一些传统机器学习算法
Stars: ✭ 129 (-2.27%)
Mutual labels:  clustering
Mlr
Machine Learning in R
Stars: ✭ 1,542 (+1068.18%)
Mutual labels:  clustering
Rectorch
rectorch is a pytorch-based framework for state-of-the-art top-N recommendation
Stars: ✭ 121 (-8.33%)
Mutual labels:  autoencoder
Community
A Python implementation of Girvan-Newman algorithm
Stars: ✭ 125 (-5.3%)
Mutual labels:  clustering
Ml Dl Scripts
The repository provides usefull python scripts for ML and data analysis
Stars: ✭ 119 (-9.85%)
Mutual labels:  clustering
View All Similar Projects ➔

pt-dec

Build Status codecov Codacy Badge

PyTorch implementation of a version of the Deep Embedded Clustering (DEC) algorithm. Compatible with PyTorch 1.0.0 and Python 3.6 or 3.7 with or without CUDA.

This follows (or attempts to; note this implementation is unofficial) the algorithm described in "Unsupervised Deep Embedding for Clustering Analysis" of Junyuan Xie, Ross Girshick, Ali Farhadi (https://arxiv.org/abs/1511.06335).

Examples

An example using MNIST data can be found in the examples/mnist/mnist.py which achieves around 85% accuracy.

Here is an example confusion matrix, true labels on y-axis and predicted labels on the x-axis.

Alt text

Usage

This is distributed as a Python package ptdec and can be installed with python setup.py install after installing ptsdae from https://github.com/vlukiyanov/pt-sdae. The PyTorch nn.Module class representing the DEC is DEC in ptdec.dec, while the train function from ptdec.model is used to train DEC.

Other implementations of DEC

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].