shobrook / gnn-dtsp

MATH 490 Final Project: Approximating solutions to the decision variant of the TSP with Graph Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MATH 490 Group 7: Philip Dohm, Roger Qiu, Jonathan Shobrook

  • ./final.ipynb - Implements GNN to solve the Traveling Salesman Problem
  • ./data/figures/train_stats.png - Visualization of training/generalization statistics
  • ./data/data/pickles/ - Model training checkpoints and training/testing statisitics

To run: just run the final.ipynb file.

Description:

Tensorflow and graphs_nets implementation of solving the Traveling Salesman Problem using a Graph Neural Network.

Dependencies:

  • random
  • tensorflow
  • networkx
  • numpy
  • pandas
  • time
  • pickle
  • tsp_solver
  • graphs_nets

About

MATH 490 Final Project: Approximating solutions to the decision variant of the TSP with Graph Neural Networks


Languages

Language:HTML 76.0%Language:Jupyter Notebook 20.0%Language:Python 4.0%