YAO/YAO-optimizer
1
0
Fork 0
mirror of https://github.com/newnius/YAO-optimizer.git synced 2025-06-07 07:01:56 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-05-02 01:38:39 +08:00
parent f3cef4f20a
commit e6bc92507c
2 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
.idea/workspace.xml
View File

@ -1,7 +1,9 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ChangeListManager">
<list default="true" id="0aedafd8-e57e-462a-beda-65af0b91f3df" name="Default Changelist" comment="" />
<list default="true" id="0aedafd8-e57e-462a-beda-65af0b91f3df" name="Default Changelist" comment="">
<change beforePath="$PROJECT_DIR$/serve.py" beforeDir="false" afterPath="$PROJECT_DIR$/serve.py" afterDir="false" />
</list>
<ignored path="$PROJECT_DIR$/out/" />
<option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
<option name="SHOW_DIALOG" value="false" />
@ -45,7 +47,7 @@
<counts>
<entry key="Dockerfile" value="75" />
<entry key="md" value="104" />
<entry key="py" value="2316" />
<entry key="py" value="2330" />
<entry key="sh" value="5" />
</counts>
</usages-collector>
@ -54,7 +56,7 @@
<entry key="Bash" value="5" />
<entry key="Dockerfile" value="75" />
<entry key="Markdown" value="104" />
<entry key="Python" value="2316" />
<entry key="Python" value="2330" />
</counts>
</usages-collector>
</session>
@ -76,11 +78,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="312">
<caret line="131" column="21" lean-forward="true" selection-start-line="131" selection-start-column="21" selection-end-line="131" selection-end-column="21" />
<state relative-caret-position="259">
<caret line="186" column="56" selection-start-line="186" selection-start-column="56" selection-end-line="186" selection-end-column="56" />
<folding>
<element signature="e#18#46#0" expanded="true" />
<marker date="1588354187030" expanded="true" signature="5573:5575" ph="..." />
<marker date="1588354485880" expanded="true" signature="5573:5575" ph="..." />
</folding>
</state>
</provider>
@ -119,8 +121,8 @@
<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="57">
<caret line="82" column="26" selection-start-line="82" selection-start-column="20" selection-end-line="82" selection-end-column="26" />
<state relative-caret-position="774">
<caret line="85" lean-forward="true" selection-start-line="85" selection-end-line="85" />
<folding>
<element signature="e#0#23#0" expanded="true" />
</folding>
@ -148,6 +150,7 @@
<find>batch_size</find>
<find>predict</find>
<find>train_num</find>
<find>label_in_feature_columns</find>
</findStrings>
</component>
<component name="Git.Settings">
@ -207,7 +210,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="1588354296835" />
<property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588354652110" />
<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" />
@ -248,16 +251,15 @@
<option name="presentableId" value="Default" />
<updated>1588152877746</updated>
<workItem from="1588152880522" duration="16973000" />
<workItem from="1588319878551" duration="13956000" />
<workItem from="1588319878551" duration="14254000" />
</task>
<servers />
</component>
<component name="TimeTrackingManager">
<option name="totallyTimeSpent" value="30929000" />
<option name="totallyTimeSpent" value="31227000" />
</component>
<component name="ToolWindowManager">
<frame x="0" y="0" width="1280" height="800" extended-state="0" />
<editor active="true" />
<layout>
<window_info id="Designer" order="0" />
<window_info id="UI Designer" order="1" />
@ -358,8 +360,8 @@
</entry>
<entry file="file://$PROJECT_DIR$/model_tensorflow.py">
<provider selected="true" editor-type-id="text-editor">
<state relative-caret-position="57">
<caret line="82" column="26" selection-start-line="82" selection-start-column="20" selection-end-line="82" selection-end-column="26" />
<state relative-caret-position="774">
<caret line="85" lean-forward="true" selection-start-line="85" selection-end-line="85" />
<folding>
<element signature="e#0#23#0" expanded="true" />
</folding>
@ -368,11 +370,11 @@
</entry>
<entry file="file://$PROJECT_DIR$/serve.py">
<provider selected="true" editor-type-id="text-editor">
<state relative-caret-position="312">
<caret line="131" column="21" lean-forward="true" selection-start-line="131" selection-start-column="21" selection-end-line="131" selection-end-column="21" />
<state relative-caret-position="259">
<caret line="186" column="56" selection-start-line="186" selection-start-column="56" selection-end-line="186" selection-end-column="56" />
<folding>
<element signature="e#18#46#0" expanded="true" />
<marker date="1588354187030" expanded="true" signature="5573:5575" ph="..." />
<marker date="1588354485880" expanded="true" signature="5573:5575" ph="..." />
</folding>
</state>
</provider>

3
serve.py
View File

@ -177,13 +177,12 @@ def train_models():
train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data()
print(train_X, valid_X, train_Y, valid_Y)
print(train_X.shape[0])
if train_X.shape[0] < 500:
config.batch_size = 32
if train_X.shape[0] < 200:
config.batch_size = 16
print(train_X[:, :1], valid_X[:, :1], train_Y, valid_Y)
train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y)
lock.release()