Skip to content

Commit 893f99b

Browse files
authored
Merge pull request #221 from drj11/drj11/fix-real-labels
Fix real labels
2 parents 5bb4e49 + 75ffb2f commit 893f99b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_wrapper/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1528,7 +1528,7 @@ def preprocessReal(labels_list):
15281528
labels = []
15291529
with codecs.open(labels_list, 'r', encoding='utf-8') as list_:
15301530
for line in list_:
1531-
labels.append(int(line.rstrip('\n')))
1531+
labels.append(float(line))
15321532
elif isinstance(labels_list, list):
15331533
labels = labels_list
15341534
else:

0 commit comments

Comments
 (0)