Skip to content

Mutli-label text classification using ConvNet and graph embedding (Tensorflow implementation)

Notifications You must be signed in to change notification settings

erichan2046/multi-label-text-classification-2

 
 

Repository files navigation

graph embedding + deep learning for multi-label text classification

this projects attempts to combine:

  • graph embedding
  • ConvNet

for the purpose of multi-label text classification.

I compared three methods on stackexchange datasets, where the goal is to predict the tags of posts.

If you wan to know more, here are some slides

utility scripts

  • scripts/preprocessing_pipeline.sh: all the preprocessing, data splitting, feature extractio, etc
  • sample_random_walks.py: sample random walks on a graph
  • extract_embedding_labels.py: extract labels for embedding visualization

main scripts

  • fastxml_experiment.py: experiment for fastxml
  • kim_cnn_experiment.py: experiment for cnn
  • combined_model_experiment.py: experiment for cnn + deepwalk

About

Mutli-label text classification using ConvNet and graph embedding (Tensorflow implementation)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 61.8%
  • Python 38.1%
  • Shell 0.1%