Small fix for saving to csv
Some checks failed
s434704-training/pipeline/head There was a failure building this commit
Some checks failed
s434704-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
b73ac931cb
commit
b478c680bb
@ -36,9 +36,9 @@ film_train, film_test = train_test_split(film_data, test_size=1 - train_ratio)
|
||||
|
||||
film_valid, film_test = train_test_split(film_test, test_size=test_ratio/(test_ratio + validation_ratio))
|
||||
|
||||
pd.to_csv(film_train, 'MoviesOnStreamingPlatforms_updated.train')
|
||||
pd.to_csv(film_test, 'MoviesOnStreamingPlatforms_updated.test')
|
||||
pd.to_csv(film_valid, 'MoviesOnStreamingPlatforms_updated.valid')
|
||||
film_train.to_csv('MoviesOnStreamingPlatforms_updated.train')
|
||||
film_test.to_csv('MoviesOnStreamingPlatforms_updated.test')
|
||||
film_valid.to_csv('MoviesOnStreamingPlatforms_updated.valid')
|
||||
|
||||
# Statystki głównego zbioru i podzbiorów
|
||||
for i, data_set in enumerate([film_data, film_train, film_valid, film_test]):
|
||||
|
Loading…
Reference in New Issue
Block a user