List-based subset assignment

This example demonstrates how to distribute your dataset between Test and Training subsets using lists.

Import necessary libraries

import ansys.simai.core

Create lists

List the data to be used for the test set.

# CAUTION:
# All training data that are not included into the following list will be
# assigned the training subset
TEST_LIST = ["My_td_1", "My_td_2"]

Connect to the platform

Connect to the SimAI platform. Refer to the Credentials section of the documentation to adapt the connection type.

simai = ansys.simai.core.SimAIClient(organization="My_organization_name")
project = simai.projects.get(name="My_project_name")

Assign subsets

Assign a subset to each dataset (list) you created.

td_list = project.data
for td in td_list:
    if td.name in TEST_LIST:
        td.assign_subset(project, "Test")
    else:
        td.assign_subset(project, "Training")