fix script

This commit is contained in:
Alicja Szulecka 2024-04-13 19:17:29 +02:00
parent 4709691adf
commit 3ca7e62805

View File

@ -1,19 +1,19 @@
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')
forest_train = pd.read_csv('forest_train.csv')
forest_test = pd.read_csv('forest_test.csv')
forest_val = pd.read_csv('forest_val.csv')
with open('stats.txt', 'w') as f:
f.write(f"Wielkosc treningowego zbioru:\n {meteorite_train.shape}\n")
f.write(f"Wielkosc testowego zbioru:\n {meteorite_test.shape}\n")
f.write(f"Wielkosc walidacyjnego zbioru:\n {meteorite_val.shape}\n")
f.write(f"Wielkosc treningowego zbioru:\n {forest_train.shape}\n")
f.write(f"Wielkosc testowego zbioru:\n {forest_test.shape}\n")
f.write(f"Wielkosc walidacyjnego zbioru:\n {forest_val.shape}\n")
f.write(f"\Zbior Treningowy:\n")
f.write(f"{meteorite_train.describe()}\n")
f.write(f"{forest_train.describe()}\n")
f.write(f"\Zbior Testowy:\n")
f.write(f"{meteorite_test.describe()}\n")
f.write(f"{forest_test.describe()}\n")
f.write(f"\Zbior walidacyjny:\n")
f.write(f"{meteorite_val.describe()}\n")
f.write(f"{forest_val.describe()}\n")