config.py

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': 10,

    # generative_model_setting
    'num_beams': 5,
    'num_return_sequences': 3,

    # hw_classifier setting
    'max_length': 512
}

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')],
        }
    }
}