Small fix for saving to csv
Some checks failed
s434704-training/pipeline/head There was a failure building this commit

This commit is contained in:
Wojciech Jarmosz 2021-05-14 01:01:22 +02:00
parent b73ac931cb
commit b478c680bb

View File

@ -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]):