diff --git a/get_stats.py b/get_stats.py index e8ca59f..4b851cc 100644 --- a/get_stats.py +++ b/get_stats.py @@ -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") \ No newline at end of file + f.write(f"{forest_val.describe()}\n") \ No newline at end of file