docker-v3

This commit is contained in:
s434732 2021-04-11 09:54:05 +02:00
parent 176b30ef42
commit a63041cfc1
2 changed files with 11 additions and 11 deletions

View File

@ -21,7 +21,12 @@ train, test = train_test_split(results, test_size= 1 - 0.6)
valid, test = train_test_split(test, test_size=0.5) valid, test = train_test_split(test, test_size=0.5)
train.to_csv("train.csv", index=False)
valid.to_csv("valid.csv",index=False)
test.to_csv("test.csv",index=False)
print("All data: ", results.size)
print("Train size: ", train.size)
print("Test size: ", test.size)
print("Validate size: ", valid.size)
print(results.describe(include='all'))
# sprawdzenie czy cały dataset oraz podział na podzbiory jest równy
print(train.size+test.size+valid.size)

View File

@ -21,11 +21,6 @@ train, test = train_test_split(results, test_size= 1 - 0.6)
valid, test = train_test_split(test, test_size=0.5) valid, test = train_test_split(test, test_size=0.5)
print("All data: ", results.size) train.to_csv("train.csv", index=False)
print("Train size: ", train.size) valid.to_csv("valid.csv",index=False)
print("Test size: ", test.size) test.to_csv("test.csv",index=False)
print("Validate size: ", valid.size)
print(results.describe(include='all'))
# sprawdzenie czy cały dataset oraz podział na podzbiory jest równy
print(train.size+test.size+valid.size)