AutoTrain 文档

表格分类/回归

您正在查看 版本,它需要从源代码安装. 如果您希望使用常规的 pip 安装,请查看最新的稳定版本 (v0.8.8).
Hugging Face's logo
加入 Hugging Face 社区

并获取增强文档体验

开始

表格分类/回归

使用 AutoTrain,您可以轻松地训练模型来对表格数据进行分类或回归。您只需从模型列表中选择一个模型并上传您的数据集即可。参数调整会自动完成。

模型

以下模型可用于表格分类/回归。

  • xgboost
  • random_forest
  • ridge
  • logistic_regression
  • svm
  • extra_trees
  • gradient_boosting
  • adaboost
  • decision_tree
  • knn

数据格式

id,category1,category2,feature1,target
1,A,X,0.3373961604172684,1
2,B,Z,0.6481718720511972,0
3,A,Y,0.36824153984054797,1
4,B,Z,0.9571551589530464,1
5,B,Z,0.14035078041264515,1
6,C,X,0.8700872583584364,1
7,A,Y,0.4736080452737105,0
8,C,Y,0.8009107519796442,1
9,A,Y,0.5204774795512048,0
10,A,Y,0.6788795301189603,0
.
.
.

您的 CSV 数据集必须包含两列:idtarget

< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.