Skip to content

Commit 31e5d83

Browse files
author
aditya tewari
committed
adding -1 to label to load from matlab data labels 1 onwards to 0 onwards, on lienet loss = nan still.
1 parent 11e9efe commit 31e5d83

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/shared/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,4 +42,4 @@ def load_matlab_data(key, data_dir, *folders):
4242
for filename in glob.glob(os.path.join(*folders)):
4343
examples.append(scipy.io.loadmat(filename)[key])
4444
labels.append(int(filename.split("/")[-2]))
45-
return np.stack(examples), np.array(labels)
45+
return np.stack(examples), np.array(labels) - 1

0 commit comments

Comments
 (0)