INNER CODE UNIT · Python

split_inputs_output

bansalkanav/Ultimate-Data-Science-Toolkit---From-Python-Basics-to-GenerativeAI · Module 5 - MLOPs/4. Orchestrate ML Pipeline/my_workflow_script.py:20

def split_inputs_output(data, inputs, output):
    """
    Split features and target variables.
    """
    X = data[inputs]
    y = data[output]
    return X, y
	

@task
def split_train_test(X, y, test_size=0.25, random_state=0):
    """
    Split data into train and test sets.
    """
    return train_test_split(X, y, test_size=test_size, random_state=random_state)
	
	
@task

View source record →