Browse Source

[Update] Change an excepetion output msg in DataLoader.

v0.2.x
jajupmochi 4 years ago
parent
commit
d61db07d58
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      gklearn/dataset/file_managers.py

+ 2
- 1
gklearn/dataset/file_managers.py View File

@@ -332,7 +332,8 @@ class DataLoader():
content_targets = ga.read().splitlines() # targets (regression) content_targets = ga.read().splitlines() # targets (regression)
targets = [int(i) for i in content_targets] targets = [int(i) for i in content_targets]
else: else:
raise Exception('Can not find targets file. Please make sure there is a "', ds_name, '_graph_labels.txt" or "', ds_name, '_graph_attributes.txt"', 'file in your dataset folder.')
exp_msg = 'Can not find targets file. Please make sure there is a "', ds_name, '_graph_labels.txt" or "', ds_name, '_graph_attributes.txt"', 'file in your dataset folder.'
raise Exception(exp_msg)
if class_label_map is not None: if class_label_map is not None:
targets = [class_label_map[t] for t in targets] targets = [class_label_map[t] for t in targets]




Loading…
Cancel
Save