Esempio n. 1
0
from algorithms.Model1 import Model1
from scrapers.scraper import Scraper
import pickle
import os
import numpy as np

from settings import settings

if __name__ == "__main__":
    print("Starting altomfotball.no scraper!")

    data = None
    if not settings["use_pickle_model"]:
        scraper = Scraper.scrape()
        pickle.dump(scraper, open(os.path.join(os.getcwd(), "model.p"), "wb"))
        data = scraper
    else:
        data = pickle.load(open(os.path.join(os.getcwd(), "model.p"), "rb"))

    model = Model1()

    X_train = []
    Y_train = []
    X_test = []
    Y_test = []

    train_years = [x for x in range(2016, 2017)]
    test_years = [x for x in range(2017, 2018)]

    for year, season in data.items():
        if year not in train_years and year not in test_years: