-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
21 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1 +1,21 @@ | ||
# Attention-based-GNN-reinforcement-learning-for-Knapsack-Problem | ||
# Attention-based-GNN-reinforcement-learning-for-Knapsack-Problem | ||
|
||
Repo for knapsack problem solution based on reinforcement learning (Attention Based Graph Neural Network) | ||
|
||
## To run: | ||
Simply execute the jupyter notebook (main_knapsack) | ||
|
||
## Notes: | ||
I have checked proper execution for 20 nodes (items-values pairs). | ||
|
||
## TO_Do: | ||
1. Change the names of files and function suitable for knapsack | ||
This is modified version of original vrp repo, so I have kept the changes minimal as they wont affect functionality, | ||
but might be confusing. SInly follow the jupter notebook. | ||
|
||
## Further Notes: | ||
This is modified version of original repo .[src](https://arxiv.org/abs/1803.08475) | ||
Changes: | ||
1. Modify VRP to Knapsack Problem | ||
2. Data Generation suitable for knapsack problem (This could be modified by the use as needed to fit for given problem) | ||
3. Loss for the selected item_value pair |