Commit 442bf104 authored by 张彦钊's avatar 张彦钊

add print

parent c91b272f
......@@ -8,8 +8,8 @@ import utils
import warnings
from multiprocessing import Pool
from config import *
import time
import json
from sklearn.preprocessing import MinMaxScaler
def test_con_sql(device_id):
db = pymysql.connect(host='rm-m5e842126ng59jrv6.mysql.rds.aliyuncs.com', port=3306, user='doris',
......@@ -119,8 +119,11 @@ def predict(queue_name, x_list):
def save_result(queue_name, x_list):
score_df = pd.read_csv(DIRECTORY_PATH + "result/output{0}_{1}.csv".format(device_id,queue_name), header=None)
score_df = score_df.rename(columns={0: "score"})
mm_scaler = MinMaxScaler()
score_df["score"] = mm_scaler.fit_transform(score_df[["score"]])
score_df["cid"] = x_list
score_df = score_df.sort_values(by="score", ascending=False)
print("概率前十行:")
print(score_df)
return score_df
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment