def get_exist_list():
exist_tup = cursor.fetchall()
exists = list()
for a_item in exist_tup:
exists.append(a_item[0])
return exists
cursor.execute(
'''UPDATE UpdateFilm SET linear_predict=%s,linear_test=%s,
lasso_predict=%s,lasso_test=%s,knn_predict=%s,knn_test=%s,poly_predict=%s,poly_test=%s
WHERE imdb_filmID = %s''',
(scoredict['linear_predict'], scoredict['linear_test'],
scoredict['lasso_predict'], scoredict['lasso_test'],
scoredict['knn_predict'], scoredict['knn_test'],
scoredict['poly_predict'], scoredict['poly_test'], filmid))
db.commit()
cursor.execute('SELECT click_times,gross '
'FROM FilmDB,TrailerClick '
'WHERE FilmDB.imdb_filmID=TrailerClick.imdb_filmID '
'AND gross>4*TrailerClick.click_times')
clicks = cursor.fetchall()
X_R1 = list()
y_R1 = list()
for click_time, gross in clicks:
X_R1.append(click_time)
y_R1.append(gross)
X_data = np.array(X_R1).reshape(-1, 1)
y_data = np.array(y_R1).reshape(-1, 1)
X_train, X_test, y_train, y_test = train_test_split(X_data,
y_data,
test_size=0.2,
random_state=4)
cursor.execute(
'SELECT TrailerClick.imdb_filmID,max(click_times) FROM UpdateFilm,TrailerClick WHERE TrailerClick.imdb_filmID=UpdateFilm.imdb_filmID GROUP BY UpdateFilm.imdb_filmID'
)
parameters = cursor.fetchall()