luozachary / drl-rec

Deep reinforcement learning for recommendation system
177 stars 52 forks source link

ValueError: cannot reshape array of size 11 into shape (1,12,30) #16

Open gangxu822 opened 1 year ago

gangxu822 commented 1 year ago

Traceback (most recent call last): File "buildestimator.py", line 398, in main(args) File "build_estimator.py", line 358, in main env = Simulator() File "/dockerdata/danielgxu/sucai_paoma/drl-rec-master/simulator.py", line 15, in init self.init_state = self.reset() File "/dockerdata/danielgxu/sucai_paoma/drl-rec-master/simulator.py", line 20, in reset init_state = np.array(self.data['state_float'].sample(1).values[0]).reshape((1, 12, 30)) ValueError: cannot reshape array of size 11 into shape (1,12,30)