IK.py 文件源码

def base_model():
     model = Sequential()
     model.add(Dense(32, input_dim=7, init='normal', activation='relu'))
     model.add(Dense(64, init='normal', activation='relu'))
     model.add(Dense(128, init='normal', activation='relu'))
     model.add(Dense(32, init='normal', activation='relu'))
     model.add(Dense(6, init='normal'))
     model.compile(loss='mean_absolute_error', optimizer = 'adam')
     return model