Project-AutoML/automl/data.py [26:37]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
PATH_ERROR_MESSAGE = (
    "data_path only support csv data or directory contained train.csv and test.csv"
)


def load_data(data_path, label_column, test_size=0.25, random_state=1):
    if os.path.isdir(data_path):
        train_path = os.path.join(data_path, "train.csv")
        test_path = os.path.join(data_path, "test.csv")
        assert os.path.exists(train_path) and os.path.exists(
            test_path
        ), PATH_ERROR_MESSAGE
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



Project-BasicAlgorithm/core/data.py [23:34]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
PATH_ERROR_MESSAGE = (
    "data_path only support csv data or directory contained train.csv and test.csv"
)


def load_data(data_path, label_column, test_size=0.25, random_state=1):
    if os.path.isdir(data_path):
        train_path = os.path.join(data_path, "train.csv")
        test_path = os.path.join(data_path, "test.csv")
        assert os.path.exists(train_path) and os.path.exists(
            test_path
        ), PATH_ERROR_MESSAGE
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -