bfwl2021
/
openGauss-server
forked from huawei/openGauss-server
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!1018 Fix issue: Fix some bugs on the xtuner 

Merge pull request !1018 from wangtq/2.0.0
This commit is contained in:
opengauss-bot 2021-06-13 14:50:41 +08:00 committed by Gitee
parent d2cc245629 6bf1fb2f9c
commit 8c0b5d2d25
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/gausskernel/dbmind/tools/xtuner/tuner/algorithms/rl_agent.py
View File

@ -13,10 +13,10 @@ MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details.
"""
from keras import Model
from keras import Sequential
from keras.layers import Flatten, Dense, Activation, Input, Concatenate
from keras.optimizers import Adam
from tensorflow.keras import Model
from tensorflow.keras import Sequential
from tensorflow.keras.layers import Flatten, Dense, Activation, Input, Concatenate
from tensorflow.keras.optimizers import Adam
from rl.agents import DDPGAgent
from rl.agents import DQNAgent
from rl.memory import SequentialMemory
@ -47,8 +47,6 @@ class RLAgent:
model.add(Activation('relu'))
model.add(Dense(16))
model.add(Activation('relu'))
model.add(Dense(16))
model.add(Activation('relu'))
model.add(Dense(nb_actions, activation='linear'))
# build alg
@ -70,8 +68,6 @@ class RLAgent:
actor.add(Activation('relu'))
actor.add(Dense(16))
actor.add(Activation('relu'))
actor.add(Dense(16))
actor.add(Activation('relu'))
actor.add(Dense(nb_actions))
actor.add(Activation('sigmoid'))
@ -84,8 +80,6 @@ class RLAgent:
x = Activation('relu')(x)
x = Dense(32)(x)
x = Activation('relu')(x)
x = Dense(32)(x)
x = Activation('relu')(x)
x = Dense(1)(x)
x = Activation('linear')(x)
critic = Model(inputs=[action_input, observation_input], outputs=x)