Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions ai/Kconfig
Original file line number Diff line number Diff line change
Expand Up @@ -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
70 changes: 70 additions & 0 deletions ai/LiME-TM/Kconfig
Original file line number Diff line number Diff line change
@@ -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
27 changes: 27 additions & 0 deletions ai/LiME-TM/package.json
Original file line number Diff line number Diff line change
@@ -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"
}
]
}
Loading