MCPcopy Create free account
hub / github.com/Modulus-Labs/RockyBot / process_hour_day_week_task

Function process_hour_day_week_task

pytorch-model/process_dataset.py:404–420  ·  view source on GitHub ↗
(idx_to_field_data, idx_to_field_labels, data_features, labels)

Source from the content-addressed store, hash-verified

402
403
404def process_hour_day_week_task(idx_to_field_data, idx_to_field_labels, data_features, labels):
405
406 # --- Doing a 10 to 1 (random) split ---
407 val_indices = np.random.choice(range(len(data_features)),
408 int(len(data_features) / 10),
409 replace=False)
410
411 # --- Extract elements ---
412 val_features = data_features[val_indices]
413 val_labels = labels[val_indices]
414
415 train_mask = np.ones(len(data_features), bool)
416 train_mask[val_indices] = False
417 train_features = data_features[train_mask]
418 train_labels = labels[train_mask]
419
420 return train_features, train_labels, val_features, val_labels, idx_to_field_data, idx_to_field_labels
421
422
423def process_save_data(train_features,

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected