diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 60e86cb..99e1605 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -45,7 +45,7 @@
         <counts>
           <entry key="Dockerfile" value="75" />
           <entry key="md" value="104" />
-          <entry key="py" value="2269" />
+          <entry key="py" value="2291" />
           <entry key="sh" value="5" />
         </counts>
       </usages-collector>
@@ -54,7 +54,7 @@
           <entry key="Bash" value="5" />
           <entry key="Dockerfile" value="75" />
           <entry key="Markdown" value="104" />
-          <entry key="Python" value="2269" />
+          <entry key="Python" value="2291" />
         </counts>
       </usages-collector>
     </session>
@@ -76,11 +76,11 @@
       <file pinned="false" current-in-tab="true">
         <entry file="file://$PROJECT_DIR$/serve.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="154">
-              <caret line="37" column="27" selection-start-line="37" selection-start-column="27" selection-end-line="37" selection-end-column="27" />
+            <state relative-caret-position="283">
+              <caret line="132" column="20" selection-start-line="132" selection-start-column="20" selection-end-line="132" selection-end-column="20" />
               <folding>
                 <element signature="e#18#46#0" expanded="true" />
-                <marker date="1588353097067" expanded="true" signature="5560:5562" ph="..." />
+                <marker date="1588353677501" expanded="true" signature="5592:5594" ph="..." />
               </folding>
             </state>
           </provider>
@@ -119,7 +119,7 @@
       <file pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/model_tensorflow.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="157">
+            <state relative-caret-position="-783">
               <caret line="26" column="45" selection-start-line="26" selection-start-column="35" selection-end-line="26" selection-end-column="45" />
               <folding>
                 <element signature="e#0#23#0" expanded="true" />
@@ -146,6 +146,7 @@
       <find>usecol</find>
       <find>valid_data_rate</find>
       <find>batch_size</find>
+      <find>predict</find>
     </findStrings>
   </component>
   <component name="Git.Settings">
@@ -205,7 +206,7 @@
   <component name="PropertiesComponent">
     <property name="WebServerToolWindowFactoryState" value="false" />
     <property name="aspect.path.notification.shown" value="true" />
-    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588353255401" />
+    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588353307531" />
     <property name="go.gopath.indexing.explicitly.defined" value="true" />
     <property name="nodejs_interpreter_path.stuck_in_default_project" value="undefined stuck path" />
     <property name="nodejs_npm_path_reset_for_default_project" value="true" />
@@ -246,12 +247,12 @@
       <option name="presentableId" value="Default" />
       <updated>1588152877746</updated>
       <workItem from="1588152880522" duration="16973000" />
-      <workItem from="1588319878551" duration="12998000" />
+      <workItem from="1588319878551" duration="13536000" />
     </task>
     <servers />
   </component>
   <component name="TimeTrackingManager">
-    <option name="totallyTimeSpent" value="29971000" />
+    <option name="totallyTimeSpent" value="30509000" />
   </component>
   <component name="ToolWindowManager">
     <frame x="0" y="0" width="1280" height="800" extended-state="0" />
@@ -355,7 +356,7 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/model_tensorflow.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="157">
+        <state relative-caret-position="-783">
           <caret line="26" column="45" selection-start-line="26" selection-start-column="35" selection-end-line="26" selection-end-column="45" />
           <folding>
             <element signature="e#0#23#0" expanded="true" />
@@ -365,11 +366,11 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/serve.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="154">
-          <caret line="37" column="27" selection-start-line="37" selection-start-column="27" selection-end-line="37" selection-end-column="27" />
+        <state relative-caret-position="283">
+          <caret line="132" column="20" selection-start-line="132" selection-start-column="20" selection-end-line="132" selection-end-column="20" />
           <folding>
             <element signature="e#18#46#0" expanded="true" />
-            <marker date="1588353097067" expanded="true" signature="5560:5562" ph="..." />
+            <marker date="1588353677501" expanded="true" signature="5592:5594" ph="..." />
           </folding>
         </state>
       </provider>
diff --git a/serve.py b/serve.py
index 3984c0b..420e40e 100644
--- a/serve.py
+++ b/serve.py
@@ -117,10 +117,10 @@ class Data:
 		test_x = [feature_data[self.start_num_in_test + i * self.config.time_step: self.start_num_in_test + (
 				i + 1) * self.config.time_step]
 		          for i in range(time_step_size)]
+		print("test_x is", test_x)
 		if return_label_data:
 			label_data = self.norm_data[self.train_num + self.start_num_in_test:, self.config.label_in_feature_columns]
 			return np.array(test_x), label_data
-		print(test_x)
 		return np.array(test_x)
 
 	# add yqy