File size: 1,587 Bytes
a20f638
 
 
 
 
aa4d3b4
 
 
c1bf2c0
a20f638
 
de21fbe
 
a20f638
 
d2f6a3f
a20f638
 
d2f6a3f
 
a20f638
 
399b20e
a20f638
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
config = {
    # tokenizer
    'tokenizer_path_retrieval': 'imamnurby/bow-tokenizer-uncased',

    # model_path
    'model_path_retrieval': './library_retriever/retrieval_distillbert',
    'model_path_generative': './pattern_generator',
    'model_path_classifier': './configuration_classifier',
    'classifier_head_path': './configuration_classifier/classifier_head_rf_11nov.pkl',

    # db path
    'db_metadata_path': 'lib_to_features.csv',
    'db_constructor_path': 'lib_to_constructor.csv',

    # retrieval_model_setting
    'max_k': 1,

    # generative_model_setting
    'num_beams': 1,
    'num_return_sequences': 1,

    # hw_classifier setting
    'max_length': 128
}

classifier_class_mapping = {
    0: {'protocol': 'UART',
        'pin_connection_from_hw_to_arduino': {
            'arduino_mega': [('RX-->19, TX-->18')],
            'arduino_uno': [('RX-->0, TX-->1')]
        }
    },
    1: {'protocol': 'SPI',
        'pin_connection_from_hw_to_arduino': {
            'arduino_mega': [('SCK-->52, MOSI-->51, MISO-->50, CS-->53')],
            'arduino_uno': [('SCK-->13, MOSI-->11, MISO-->12, CS-->10')]
        }
    },
    2: {'protocol': 'I2C',
        'pin_connection_from_hw_to_arduino': {
            'arduino_mega': [('SDA-->20, SCL-->21')],
            'arduino_uno': [('SDA-->A4, SCL-->A5')],
        }
    },
    3: {'protocol': 'Explicit declaration',
        'pin_connection_from_hw_to_arduino': {
            'arduino_mega': [('DATA-->any digital or analog pins')],
            'arduino_uno': [('DATA-->any digital or analog pins')],
        }
    }
}