docker-v3
This commit is contained in:
parent
e36cdc9e13
commit
8b605c962c
10
skrypt.py
10
skrypt.py
@ -21,11 +21,7 @@ train, test = train_test_split(results, test_size= 1 - 0.6)
|
||||
|
||||
valid, test = train_test_split(test, test_size=0.5)
|
||||
|
||||
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'))
|
||||
train.to_csv("train.csv", index=False)
|
||||
valid.to_csv("valid.csv",index=False)
|
||||
test.to_csv("test.csv",index=False)
|
||||
|
||||
# sprawdzenie czy cały dataset oraz podział na podzbiory jest równy
|
||||
print(train.size+test.size+valid.size)
|
||||
|
Loading…
Reference in New Issue
Block a user