Beispiel #1
0
def test(net: EMGClassifier, gesture_amount, gesture_indices: list):
    test_set = CSLDataset(gesture=gesture_amount,
                          frame_x=True,
                          sequence_len=10,
                          train=False)

    avg_score = net.test_model(gesture_indices, test_set)
    print('test accuracy: {:.4f}'.format(avg_score))
    return net
Beispiel #2
0
def test(net: EMGClassifier, gesture_indices: list):
    test_set = CapgDataset(gestures_label_map=net.gesture_map,
                           sequence_len=1,
                           frame_x=True,
                           gesture_list=gesture_indices,
                           train=False)

    avg_score, matrix = net.test_model(gesture_indices, test_set)
    print('test accuracy: {:.4f}'.format(avg_score))
    return matrix
Beispiel #3
0
def test(net: EMGClassifier, gesture_indices: list):
    if net.hyperparamters['dataset'] == 'capg':
        test_set = CapgDataset(gestures_label_map=net.gesture_map,
                               sequence_len=1,
                               gesture_list=gesture_indices,
                               train=False)
    else:
        test_set = CSLDataset(gesture=8,
                              sequence_len=1,
                              train=False)

    avg_score, matrix = net.test_model(gesture_indices, test_set)
    print('test accuracy: {:.4f}'.format(avg_score))
    return matrix