diff --git a/.idea/workspace.xml b/.idea/workspace.xml index ddac956..b625821 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -203,7 +203,7 @@ - + diff --git a/serve.py b/serve.py index 2a7bb9e..7ea3455 100644 --- a/serve.py +++ b/serve.py @@ -216,7 +216,7 @@ class MyHandler(BaseHTTPRequestHandler): with open(config.train_data_path, 'r') as f: df = pd.read_csv(config.train_data_path, usecols=['job', 'model', 'time', 'utilCPU', 'utilGPU', 'pre', 'main', 'post']) - df = df.tail(config.time_step) + df = df.tail(config.time_step - 1) df = df.append(data, ignore_index=True) df.to_csv('./data/test_data.csv', index=False)