Как создать тестовый набор данных нескольких классов, используя numpy?

Question

Я хочу создать набор тестовых данных для нескольких классов, используя numpy только для задачи классификации.Например, X - это пустой массив измерения (mxn), y измерения (mx1), и скажем, что k нет.классов.Пожалуйста, помогите мне с кодом.[Здесь X представляет особенности, а y представляет метки]

Answer 1

Вы можете использовать np.random.randint как:

import numpy as np
m = 4
n = 4
k = 5
X = np.random.randint(0,2,(m,n))

X
array([[1, 1, 1, 1],
   [1, 0, 0, 1],
   [1, 1, 0, 0],
   [1, 1, 1, 1]])

y = np.random.randint(0,k,m)

y
array([3, 3, 0, 4])