diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 78a0b14..06bb346 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -47,7 +47,7 @@ - + @@ -56,7 +56,7 @@ - + @@ -88,10 +88,10 @@ - + - + @@ -214,7 +214,7 @@ - + @@ -255,16 +255,15 @@ - - @@ -376,10 +375,10 @@ - + - + diff --git a/serve.py b/serve.py index 8c28a6e..b4602f2 100644 --- a/serve.py +++ b/serve.py @@ -210,10 +210,9 @@ class MyHandler(BaseHTTPRequestHandler): } with open(config.train_data_path, 'r') as f: - q = deque(f, config.time_step - 1) - df = pd.read_csv(StringIO(''.join(q)), + df = pd.read_csv(config.train_data_path, usecols=['job', 'model', 'time', 'utilCPU', 'utilGPU', 'pre', 'main', 'post']) - + df = df.tail(config.time_step - 1) print(df) df = df.append(data, ignore_index=True) df.to_csv('./data/test_data.csv', index=False)