lawlict / EEND_loss

End-to-end diarization loss
22 stars 4 forks source link

raise HungarianError("Unable to find results. Algorithm has failed.") #3

Open speaker-lover opened 3 years ago

speaker-lover commented 3 years ago

When I input :profit_matrix = np.array([[0.47341415,0.50526756,0.5631776 ,0.55596864,0.5414672,0.5414672], [0.6781792,0.23801588,0.20288624,0.55276686,0.109804,0.109804], [0.42393875,0.370307,0.37129858,0.5507781,0.2596858,0.2596858], [0.52666223,0.57174313,0.5873437,0.6438369,0.5166061,0.5166061], [0.6931475,0.6931475,0.6931475,0.6931475,0.6931475,0.6931475], [0.6931475,0.6931475,0.6931475,0.6931475,0.6931475,0.6931475]]) a error will appear as follows. File "models.py", line 41, in minimize_loss H.calculate(np_loss_mat) File "hungarian.py", line 151, in calculate raise HungarianError("Unable to find results. Algorithm has failed.") I do not know the reasons. Look forward to your reply.

lawlict commented 3 years ago

Hi @yfchen97 , I have fixed the problem by replacing Hungurian.py with scipy.optimize.linear_sum_assignment. You can pull the latest code and have a try!