DataTransfomation.py

# Import Pandas
import pandas as pd

file_name = 'India_in_Tests.csv'
fields = ['Result','Toss','Bat']

# Read the CSV file
df = pd.read_csv(file_name,skipinitialspace=True,usecols=fields)
df.to_csv('India_in_Tests_Filter.csv',index=False)

# Convert features and labels into digits
df_replace = df.replace(['lost','draw','won','1st','2nd'],[-1,0,1,-1,1])
dataset_length = len(df_replace)

# 65% of training data
ratio = 0.65

"""
First 65% Data -- For Training
"""
train_data_df = df_replace[:int(dataset_length*ratio)] 

# 35% Data - For Testing
test_data_df = df_replace[-(1-int(dataset_length*ratio)):] 

# Create Respected CSV
train_data_df.to_csv('Trained_Data.csv',index=False)
test_data_df.to_csv('Test.csv',index=False)

***
# 75% of training data
ratio = 0.75

"""
First 75% Data -- For Training
"""
train_data_df = df_replace[:int(dataset_length*ratio)] 

# 35% Data - For Testing
test_data_df = df_replace[-(1-int(dataset_length*ratio)):] 

# Create Respected CSV
train_data_df.to_csv('Trained_Data.csv',index=False)
test_data_df.to_csv('Test.csv',index=False)
***