So I am trying to train a neural network for the knapsack problem and so I have created an input pickle file that is a list of 5,000 elements. Each element has 100 weights a