This commit is contained in:
s444501 2022-04-02 22:53:10 +02:00
parent 5d210c0e6a
commit 5d564ac423

View File

@ -15,12 +15,16 @@ def column_stat(analyzed_set, column_name):
rating_median = analyzed_set[column_name].median() rating_median = analyzed_set[column_name].median()
rating_std = round(analyzed_set[column_name].std(), 3) rating_std = round(analyzed_set[column_name].std(), 3)
print(f"Dla kolumny '{column_name}':") output = ''
print(f"Minimum: {rating_min}")
print(f"Maximum: {rating_max}") output += f"Dla kolumny '{column_name}':\n"
print(f"Średnia: {rating_mean}") output += f"Minimum: {rating_min}\n"
print(f"Mediana: {rating_median}") output += f"Maximum: {rating_max}\n"
print(f"Odchylenie standardowe: {rating_std}") output += f"Średnia: {rating_mean}\n"
output += f"Mediana: {rating_median}\n"
output += f"Odchylenie standardowe: {rating_std}\n"
return output
# Pobieranie danych # Pobieranie danych
@ -48,25 +52,30 @@ d_test.to_csv('d_test.csv', index=False)
d_dev.to_csv('d_dev.csv', index=False) d_dev.to_csv('d_dev.csv', index=False)
# Statystyki # Statystyki
print(f"Wielkość całego zbioru: {disney.shape[0]}\n" temp = ''
f"Inne statystyki:") temp += f"Wielkość całego zbioru: {disney.shape[0]}\n"
column_stat(disney, 'Rating') temp += f"Inne statystyki:\n"
print('') temp += column_stat(disney, 'Rating')
temp += '\n'
print(f"Wielkość zbioru trenującego: {d_train.shape[0]}\n" temp += f"Wielkość zbioru trenującego: {d_train.shape[0]}\n"
f"Inne statystyki:") temp += f"Inne statystyki:\n"
column_stat(d_train, 'Rating') temp += column_stat(d_train, 'Rating')
print('') temp += '\n'
print(f"Wielkość zbioru walidującego: {d_dev.shape[0]}\n" temp += f"Wielkość zbioru walidującego: {d_dev.shape[0]}\n"
f"Inne statystyki:") temp += f"Inne statystyki:\n"
column_stat(d_dev, 'Rating') temp += column_stat(d_dev, 'Rating')
print('') temp += '\n'
print(f"Wielkość zbioru testowego: {d_test.shape[0]}\n" temp += f"Wielkość zbioru testowego: {d_test.shape[0]}\n"
f"Inne statystyki:") temp += f"Inne statystyki:\n"
column_stat(d_test, 'Rating') temp += column_stat(d_test, 'Rating')
print('') temp += '\n'
with open('stats.txt', 'w+', encoding="utf-8") as f:
print(temp)
f.write(temp)
# Rozkład ocen dla każdego oddziału # Rozkład ocen dla każdego oddziału
try: try:
@ -75,3 +84,4 @@ try:
plt.show() plt.show()
except: except:
print("Error drawing hist plot (Powinno działać w Pycharmie)") print("Error drawing hist plot (Powinno działać w Pycharmie)")