Skip to content

sebdei/deep_nlp_clustering

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep NLP Clustering implementation with an LSTM-Autoencoder

Groundwork by Xie et. al

Pretrain an autoencoder, cut of the encoder part and use the latent features for clustering assignment hardening.

GPU

Tensorflow-GPU support with Docker

Build Tensorflow Docker

docker build -t sebdei/tensorflow-gpu ./gpu

Run Docker with mounted volumes

docker run --gpus all -it  \
--volume ${PWD}/src:/src \
sebdei/tensorflow-gpu:latest

Check GPU activity

nvidia-smi -l 1

Python Docker

Start Python docker

docker-compose run python

Go directly to python console with python

Jupyter

Start docker

docker-compose run --service-ports jupyter