cerebral.features.create_datasets(data: DataFrame, targets: list[str], train: list | DataFrame = [], test: list | DataFrame = [])[source]

Separates the total data set of alloy compositions into training and test subsets.

Parameters:
data: DataFrame

The dataset of alloy compositions.

targets: list[str]

The features to be modelled by the neural network.

train: list | DataFrame = []

If provided, a preselected subset of data to be used for training.

test: list | DataFrame = []

If provided, a preselected subset of data to be used for testing.


Last update: Feb 09, 2023