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