13hannes11/UU_NCML_Project
1
0
Fork 0
mirror of https://github.com/13hannes11/UU_NCML_Project.git synced 2024-09-03 20:50:59 +02:00
Code Issues Projects Releases Wiki Activity

Bug fix: data used for plotting is corrected

Browse Source
This commit is contained in:
Deepthi Pathare
2021-05-04 19:18:37 +02:00
parent e302ec0956
commit 272e5721f3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
VotingAnalysis.py
View File

@@ -250,7 +250,7 @@ prediction = sofmnet.predict(X)
print(f'prediction: {prediction}')
# converting to x and y coordinates
ys, xs = np.unravel_index(np.argmax(X, axis=1), (h, w))
ys, xs = np.unravel_index(np.argmax(prediction, axis=1), (h, w))
# plotting mps
plot_mps(data[:,0], xs, ys, data[:,1])
@@ -305,7 +305,7 @@ prediction = sofmnet.predict(X)
print(f'prediction: {prediction}')
# converting to x and y coordinates
ys, xs = np.unravel_index(np.argmax(X, axis=1), (h, w))
ys, xs = np.unravel_index(np.argmax(prediction, axis=1), (h, w))
# plotting mps
plot_mps(data[:,0], xs, ys, data[:,1])