TF Decision Trees
T
TF Decision Trees
由 keras-io 开发
基于TensorFlow的梯度提升树模型,用于结构化数据的二元分类任务。
下载量 45
发布时间 : 3/2/2022
模型简介
该模型使用TensorFlow实现梯度提升树算法,专门用于处理结构化数据的二元分类问题。通过自定义二元目标编码器对分类特征进行编码,并结合决策森林模型进行预测。
模型特点
自定义二元目标编码器
实现自定义的二元目标编码器作为Keras预处理层,根据目标值的共现情况对分类特征进行编码。
高效处理结构化数据
专门针对包含数值和分类变量的结构化数据设计,能有效处理41个特征的数据集。
高准确率
在人口普查收入数据集上达到96.57%的准确率,验证损失仅为0.227394。
模型能力
结构化数据分类
二元分类预测
特征编码转换
使用案例
社会经济分析
收入水平预测
预测个人年收入是否超过5万美元
准确率96.57%
人口统计
人口普查数据分析
分析人口普查数据中的收入分布模式
精选推荐AI模型
Llama 3 Typhoon V1.5x 8b Instruct
专为泰语设计的80亿参数指令模型,性能媲美GPT-3.5-turbo,优化了应用场景、检索增强生成、受限生成和推理任务
大型语言模型
Transformers 支持多种语言

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一个基于SODA数据集训练的超小型对话模型,专为边缘设备推理设计,体积仅为Cosmo-3B模型的2%左右。
对话系统
Transformers 英语

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基于RoBERTa架构的中文抽取式问答模型,适用于从给定文本中提取答案的任务。
问答系统 中文
R
uer
2,694
98