ValueError: cannot reshape array of size 49639 into shape (148917,1)

[1zylzrb1]

Traceback (most recent call last):
File "/home/zyl/Projects/Heterodata/FMG/mf_features_generator.py", line 165, in
run_all_yelp()
File "/home/zyl/Projects/Heterodata/FMG/mf_features_generator.py", line 95, in run_all_yelp
run(path_str)
File "/home/zyl/Projects/Heterodata/FMG/mf_features_generator.py", line 49, in run
U,V = mf.run()
^^^^^^^^
File "/home/zyl/Projects/Heterodata/FMG/mf.py", line 135, in run
obs = omega.copy().data.astype(np.float64).reshape(self.train_num, 1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: cannot reshape array of size 49639 into shape (148917,1)
When I run" python mf_features_generator.py yelp-50k all 1", this issue appears. Can you help me out? Thank you.