AI & MACHINE LEARNING PROGRAM • LEVEL 01 — AI & ML FOUNDATIONS

Create a Train-test Split with Python

Learn create a train-test split with python with a short, executable Python example.

IntermediateTraining dataTesting data

PROBLEM UNDERSTANDING

Input and expected output

Sample input
No input required
Sample output
[0, 1, 2, 3, 4, 5, 6, 7] [8, 9]

COMPLETE PYTHON PROGRAM

Complete Python implementation

ai-ml-01-train-test-split.py
Open in compiler
data = list(range(10))
split = int(len(data) * 0.8)
print(data[:split], data[split:])

GUIDED CODE TOUR • NOT LIVE EXECUTION

Study the program line by line

Use the real compiler button above to run and debug with different inputs.

CURRENT STEP

Select Start to walk through the important lines.

SELECTED LINE

No line selected

EXPECTED OUTPUT FOR THE SAMPLE

[0, 1, 2, 3, 4, 5, 6, 7] [8, 9]
0%Step 0 of 0

PROGRAM EXPLANATION

Algorithm and explanation

  1. Initialize the sample values used to create a train-test split.
  2. Apply Training data and Testing data to compute the required result.
  3. Display the result for create a train-test split and compare it with the documented sample output.

This example of create a train-test split computes the result directly from the prepared sample data. It demonstrates Training data and Testing data and prints a deterministic result that can be checked against the sample output.

EFFICIENCY

Time and space complexity

Time complexity

O(n)

Auxiliary space

O(n)

DEBUGGING CHECKLIST

Common mistakes

Check this

For create a train-test split, keep the data shape and value types consistent with Training data.

Check this

Apply Training data in the same order shown by the algorithm; changing the order can change the result.

Check this

Verify the final Training data and Testing data result against the sample before trying new data.

Try it yourself

Practice: Run the program with the sample input, predict its output, and then test one boundary case of your own.