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 @@
1588152877746
-
+
-
+
-
@@ -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)