training_set = df.iloc[:800, 1:2].values
test_set = df.iloc[800:, 1:2].values