def train(net: EMGClassifier, gesture_amount):
train_set = CSLDataset(gesture=gesture_amount,
frame_x=True,
sequence_len=10,
train=True)
net.dataset = train_set
net.fit_with_dataset()
return net
def train(net: EMGClassifier, gesture_indices: list):
train_set = CapgDataset(gestures_label_map=net.gesture_map,
sequence_len=1,
frame_x=True,
gesture_list=gesture_indices,
train=True)
net.dataset = train_set
net.fit_with_dataset()
return net
def train(net: EMGClassifier, gesture_indices: list):
if net.hyperparamters['dataset'] == 'capg':
train_set = CapgDataset(gestures_label_map=net.gesture_map,
sequence_len=1,
gesture_list=gesture_indices,
train=True)
else:
train_set = CSLDataset(gesture=8,
sequence_len=1,
train=True)
net.dataset = train_set
net.fit_with_dataset()
return net