Этот код должен работать для линейных данных:
for index in range(row_data_size): Y[index] = X[index, :].sum()
, а для дерева решений:
for index in range(row_data_size): Y[index] = -1 if X[index, 0] < 0.5 else 1