#!/usr/bin/python
# -*- coding: UTF-8 -*-
import pymysql
from db.index import DataBaseHandle
from utils.probability.model import Predictor2
from utils.qq import P_get


# mysql = DataBaseHandle()
# allList = mysql.selectDb("select * from wares")
# for row in allList:
#     print(row)

# P_get()

model = Predictor2()
print(model.predict(844, 2,
                    [
                        [2012, 22176], [2013, 22352], [2014, 21302], [2015, 20386],
                        [2016, 20044], [2017, 20081], [2018, 21577]
                    ],
                    569, 2019, [22748]
                    )
      )


# gl = Predictor1()
# gl.train((936,844,2,1), [[2019, 496, 569]])
# print(gl.predict( (936,844,2,1), [[2019, 496, 569]]  ) )