Skip to content

awesome-yyh/midu_ai_coder

Repository files navigation

FeatureClassifier

模型特点:

FeatureClassifier 是一个基于全连接神经网络(Fully Connected Neural Network, FCNN)的特征分类模型。

它分为两个主要部分:

  • 特征提取器(Feature Extractor):用于从输入数据中提取高层次特征。
  • 分类头(Classifier Head):基于提取的特征进行分类任务。

使用方法

train

# 安装包
pip install -r requirements_train.txt

# 启动训练
python train.py

api

# 安装包
pip install -r requirements.txt

# 启动服务, 默认端口81
python main.py

api调用方式

url:
localhost:81/predict

# 参数:
{"features": 
    [33.5, 38.5, 37.5, 30.0, 33.5, 38.5, 41.0, 30.0, 31.5, 38.5, 41.0, 30.0, 
     31.5, 35.5, 37.0, 29.0, 31.5, 35.5, 39.0, 31.0, 33.5, 37.5, 39.0, 31.0,
     32.0, 36.5, 37.0, 28.5, 32.0, 36.5, 39.0, 30.5, 32.0, 38.5, 39.0, 32.5],
"id": "test_1"
}

# 期待的输出:
{
    "label": 7,
    "id": "test_1"
}

测试

测试api

pip install requests

python test/test_api.py

测试模型

# 
python test/test_model.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages