Fix test data

This commit is contained in:
Wojciech Jarmosz 2021-05-15 03:38:17 +02:00
parent bc20b3e4e4
commit eab03a1f84

View File

@ -11,7 +11,7 @@ pd.set_option("display.max_columns", None)
# Wczytanie danych
train_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.train")
train_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.test")
test_data = pd.read_csv("./MoviesOnStreamingPlatforms_updated.test")
# Stworzenie modelu
columns_to_use = ['Year', 'Runtime', 'Netflix']