bug fix attempt: large matrix conversion fails

This commit is contained in:
Steve Nyemba 2021-03-30 17:01:59 -05:00
parent bdd752550e
commit f65b082fb1
1 changed files with 2 additions and 1 deletions

View File

@ -603,7 +603,8 @@ class Predict(GNet):
# #
# df = pd.DataFrame(np.round(f)).astype(np.int32) # df = pd.DataFrame(np.round(f)).astype(np.int32)
candidates.append (np.round(_matrix).astype(np.int64)) # candidates.append (np.round(_matrix).astype(np.int64))
candidates.append( [np.round(row).astype(int) for row in _matrix])
# return candidates[0] if len(candidates) == 1 else candidates # return candidates[0] if len(candidates) == 1 else candidates
return candidates return candidates