| import subprocess | |
| import pandas as pd | |
| import random | |
| taskname = "spaceship-titanic" | |
| download_dir = "../env" | |
| script_dir = f"." | |
| input(f"Consent to the competition at https://www.kaggle.com/competitions/{taskname}/data; Press any key") | |
| subprocess.run(["kaggle", "competitions", "download", "-c", taskname], cwd=download_dir) | |
| subprocess.run(["unzip", "-n", f"{taskname}.zip"], cwd=download_dir) | |
| subprocess.run(["rm", f"{taskname}.zip"], cwd=download_dir) | |
| trainset = pd.read_csv(f"{download_dir}/train.csv") | |
| trainset = trainset.reset_index(drop=True) | |
| trainset.iloc[:int(len(trainset)*0.8)].to_csv(f"{download_dir}/train.csv", index=False) | |
| testset = trainset.iloc[int(len(trainset)*0.8):] | |
| testset.drop(list(trainset.keys())[1:-1], axis=1).to_csv(f"{script_dir}/answer.csv", index=False) | |
| testset = testset.drop(['Transported'], axis=1).to_csv(f"{download_dir}/test.csv", index=False) | |