s486867/ium_z486867
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
Michał Kaleta 2023-03-25 12:49:56 +01:00
commit be3e041321
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
main.py Normal file
View File

@ -0,0 +1,27 @@
from kaggle.api.kaggle_api_extended import KaggleApi
import zipfile
from sklearn.model_selection import train_test_split
import pandas as pd
pd.set_option('display.max_columns', 100)
api = KaggleApi()
api.authenticate()
api.dataset_download_files('rkiattisak/sports-car-prices-dataset', path='./data')
with zipfile.ZipFile('./data/sports-car-prices-dataset.zip', 'r') as zip_ref:
zip_ref.extractall('./data')
cars = pd.read_csv('./data/Sport car price.csv')
cars.dropna(inplace=True)
random_seed = 42
train_data, test_data = train_test_split(cars, test_size=0.2, random_state=random_seed)
train_data, dev_data = train_test_split(train_data, test_size=0.25, random_state=random_seed)
train_stats = train_data.describe(include='all')
print(f"\nTraining set statistics:\n{train_stats}")
dev_stats = dev_data.describe(include='all')
print(f"\nDevelopment set statistics:\n{dev_stats}")
test_stats = test_data.describe(include='all')
print(f"\nTest set statistics:\n{test_stats}")