armlet.data.utils module

This module contains the general data utilities.

armlet.data.utils.dataframe_safe_train_test_split(X: DataFrame, y: DataFrame, test_size: float, client_id: int | None = None)

armlet.data.utils.dataframe_train_test_split(X: DataFrame, y: DataFrame, train_size: float | None = None, test_size: float | None = None, random_state: int | None = None)