diff --git a/ai/Kconfig b/ai/Kconfig index 233db460ac..42437e0ee2 100644 --- a/ai/Kconfig +++ b/ai/Kconfig @@ -12,5 +12,6 @@ source "$PKGS_DIR/packages/ai/naxos/Kconfig" source "$PKGS_DIR/packages/ai/ncnn/Kconfig" source "$PKGS_DIR/packages/ai/r-tinymaix/Kconfig" source "$PKGS_DIR/packages/ai/llmchat/Kconfig" +source "$PKGS_DIR/packages/ai/LiME-TM/Kconfig" endmenu diff --git a/ai/LiME-TM/Kconfig b/ai/LiME-TM/Kconfig new file mode 100644 index 0000000000..03fc007897 --- /dev/null +++ b/ai/LiME-TM/Kconfig @@ -0,0 +1,70 @@ + +# Kconfig file for package ulapack +menuconfig PKG_USING_LIME_TM + select RT_USING_POSIX_FS + bool "LiME-TM: Lightning fast and Memory Efficient Machine Learning for MCU (Tsetlin Machine)" + default n + +if PKG_USING_LIME_TM + + config PKG_LIME_TM_PATH + string + default "/packages/ai/LiME-TM" + + choice + prompt "Choose exmaple" + + config LIME_TM_USING_MNIST_FS_EXAMPLE + bool "Training and testing MNIST example (File System)" + select TSETLIN_USING_PROTOBUF + help + Training a compressed Tsetlin Machine model on the MNIST dataset. + default n + + config LIME_TM_USING_MNIST_STATIC_EXAMPLE + bool "Training and testing MNIST example (Static Model)" + select TSETLIN_USING_STATIC_MODEL + select TSETLIN_MODEL_TRAINABLE + help + Training a compressed Tsetlin Machine model on the MNIST dataset. + default y + endchoice + + config TSETLIN_USING_PROTOBUF + bool "Load model from protobuf file" + help + Using proto-c to save the model as a protobuf file. + depends on LIME_TM_USING_MNIST_FS_EXAMPLE + default y + + config TSETLIN_MODEL_TRAINABLE + bool "Enable model training" + help + Enabling this option allows the model to be trained on-device. + default y + + config TSETLIN_USING_STATIC_MODEL + bool "Load model from static headers" + help + Using the tsetlin compiler to save the model as a static C header file. + depends on LIME_TM_USING_MNIST_STATIC_EXAMPLE + default y + + choice + prompt "Version" + default PKG_USING_LIME_TM_LATEST_VERSION + help + Select the package version + config PKG_USING_LIME_TM_LATEST_VERSION + bool "latest" + endchoice + + config PKG_LIME_TM_VER + string + default "latest" if PKG_USING_LIME_TM_LATEST_VERSION + + config PKG_LIME_TM_VER_NUM + hex + default 0x99999 if PKG_USING_LIME_TM_LATEST_VERSION + +endif diff --git a/ai/LiME-TM/package.json b/ai/LiME-TM/package.json new file mode 100644 index 0000000000..ddd2fe6129 --- /dev/null +++ b/ai/LiME-TM/package.json @@ -0,0 +1,27 @@ +{ + "name": "lime-tm", + "description": "Lightning fast and Memory Efficient Machine Learning for MCU (Tsetlin Machine)", + "description_zh": "嵌入式机器学习库", + "enable": "PKG_USING_LIME_TM", + "keywords": [ + "Machine Learning", + "Tsetlin Machine" + ], + "category": "ai", + "author": { + "name": "wuhanstudio", + "email": "wuhanstudio@qq.com", + "github": "wuhanstudio" + }, + "license": "MIT", + "repository": "https://github.com/wuhanstudio/tsetlin.idf", + "homepage": "https://github.com/wuhanstudio/tsetlin.idf#readme", + "site": [ + { + "version": "latest", + "URL": "https://github.com/wuhanstudio/tsetlin.idf.git", + "filename": "", + "VER_SHA": "main" + } + ] +}