Get dataset splits from comma separated string list
(args)
| 236 | |
| 237 | |
| 238 | def get_split(args): |
| 239 | """ |
| 240 | Get dataset splits from comma separated string list |
| 241 | """ |
| 242 | splits = [] |
| 243 | if args.split.find(',') != -1: |
| 244 | splits = [float(s) for s in args.split.split(',')] |
| 245 | elif args.split.find('/') != -1: |
| 246 | splits = [float(s) for s in args.split.split('/')] |
| 247 | else: |
| 248 | splits = [float(args.split)] |
| 249 | split_total = sum(splits) |
| 250 | if split_total < 1.: |
| 251 | splits.append(1-split_total) |
| 252 | while len(splits) < 3: |
| 253 | splits.append(0.) |
| 254 | splits = splits[:3] |
| 255 | if args.valid_data is not None: |
| 256 | splits[1] = 0. |
| 257 | if args.test_data is not None: |
| 258 | splits[2] = 0. |
| 259 | final_sum = sum(splits) |
| 260 | return [s/final_sum for s in splits] |
| 261 | |
| 262 | def should_split(split): |
| 263 | """ |