Python 利用make_classification 生成整数数组
时间: 2023-12-28 15:06:28 浏览: 153
make_classification函数是Scikit-learn库中用于生成分类数据集的函数,生成的数据集通常是由浮点数构成的。如果您想生成整数数组,可以使用numpy库中的random.randint函数。具体操作步骤如下:
1. 导入需要的库:
```python
from sklearn.datasets import make_classification
import numpy as np
```
2. 使用make_classification函数生成分类数据集:
```python
X, y = make_classification(n_samples=100, n_features=10, n_informative=5, n_classes=3)
```
3. 将生成的X数组中的浮点数转换为整数:
```python
X = X.astype(int)
```
4. 查看转换后的X数组:
```python
print(X)
```
这样就可以生成由整数构成的分类数据集了。
阅读全文
相关推荐









