Skip to content

Commit

Permalink
Update for numpy array
Browse files Browse the repository at this point in the history
  • Loading branch information
aPovidlo committed Aug 22, 2024
1 parent 19e732e commit 67eac10
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions fedot/core/data/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -817,15 +817,21 @@ def array_to_input_data(features_array: np.ndarray,

categorical_features = None
if categorical_idx is not None:
if isinstance(categorical_idx[0], str) and features_names is None:
raise
else:
categorical_idx = [i for i, column in enumerate(features_names) if column in set(categorical_idx)]

if isinstance(categorical_idx, list):
categorical_idx = np.array(categorical_idx)

categorical_features = features_array[:, categorical_idx]
if categorical_idx != np.array([]) and isinstance(categorical_idx[0], str) and features_names is None:
raise ValueError(
'Impossible to specify categorical features by name when the features_names are not specified'
)

if categorical_idx != np.array([]) and isinstance(categorical_idx[0], str):
categorical_idx = np.array(
[idx for idx, column in enumerate(features_names) if column in set(categorical_idx)]
)

if categorical_idx.size != 0:
categorical_features = features_array[:, categorical_idx]

data = InputData(
idx=idx,
Expand Down

0 comments on commit 67eac10

Please sign in to comment.