ium_464914/get_stats.py
Alicja Szulecka ebef6c12cc add stats
2024-04-02 22:35:43 +02:00

19 lines
688 B
Python

import pandas as pd
meteorite_train = pd.read_csv('meteorite_train.csv')
meteorite_test = pd.read_csv('meteorite_test.csv')
meteorite_val = pd.read_csv('meteorite_val.csv')
with open('stats.txt', 'w') as f:
f.write(f"Wielkość treningowego zbioru:\n {meteorite_train.shape}\n")
f.write(f"Wielkość testowego zbioru:\n {meteorite_test.shape}\n")
f.write(f"Wielkość walidacyjnego zbioru:\n {meteorite_val.shape}\n")
f.write(f"\nZbiór Treningowy:\n")
f.write(f"{meteorite_train.describe()}\n")
f.write(f"\nZbiór Testowy:\n")
f.write(f"{meteorite_test.describe()}\n")
f.write(f"\nZbiór walidacyjny:\n")
f.write(f"{meteorite_val.describe()}\n")