机器学习 | make_classification创建用于分类的数据集

Posted 2021-06-27 AI算法攻城狮

  make_classification创建用于分类的数据集，官方文档

### 创建模型
def create_model():
     
    # 生成数据
    from sklearn.datasets import make_classification
    X, y = make_classification(n_samples=10000,        # 样本个数
                               n_features=25,          # 特征个数
                               n_informative=3,        # 有效特征个数
                               n_redundant=2,          # 冗余特征个数（有效特征的随机组合）
                               n_repeated=0,           # 重复特征个数（有效特征和冗余特征的随机组合）
                               n_classes=3,            # 样本类别
                               n_clusters_per_class=1, # 簇的个数
                               random_state=0)
     
    print("原始特征维度",X.shape)
     
    # 读取数据