devsisters / neural-combinatorial-rl-tensorflow

in progress

Neural Combinatorial Optimization in Tensorflow

TensorFlow implementation of Neural Combinatorial Optimization with Reinforcement Learning.

(in progress)

Requirements

• Python 2.7
• tqdm
• TensorFlow 0.12.1

Usage

To train a model:

$ python main.py --task=tsp20 --lr_start=0.001 --min_data_length=5 --max_data_length=20
$ python main.py --task=tsp50 --lr_start=0.001 --min_data_length=5 --max_data_length=50
$ python main.py --task=tsp100 --lr_start=0.0001 --min_data_length=5 --max_data_length=100

To train a model:

$ python main.py
$ tensorboard --logdir=logs --host=0.0.0.0

To test a model:

$ python main.py --is_train=False

Results

(in progress)

Author

Taehoon Kim / @carpedm20

neural-combinatorial-rl-tensorflow's Issues

Google download error in China

Because Google is not in service somewhere, I suggest the following change in the 'data_load.py'.
`

try:

    paths = self.download_google_drive_file()

except:

    paths = ''

`

Cann't run in Mac with tensorflow 0.12.1

Traceback (most recent call last):
File "main.py", line 5, in
from trainer import Trainer
File "/Users/yobobobo/Firework/reinforce_learning/neural-combinatorial-rl-tensorflow/trainer.py", line 7, in
from model import Model
File "/Users/yobobobo/Firework/reinforce_learning/neural-combinatorial-rl-tensorflow/model.py", line 4, in
from layers import *
File "/Users/yobobobo/Firework/reinforce_learning/neural-combinatorial-rl-tensorflow/layers.py", line 7, in
LSTMCell = rnn.LSTMCell
AttributeError: 'module' object has no attribute 'LSTMCell'

RL

Where is reinforcement learning method applied ? Please tell me the difference with the code of pointer-network-tensorflow

Is this ready now?

AttributeError: module 'tensorflow.contrib.rnn' has no attribute 'LSTMCell'

Windows 10
Python 3.5.4
tensorflow-gpu 0.12.1
tqdm 4.15.0