鱼C论坛

 找回密码
 立即注册
查看: 1698|回复: 1

求大神帮忙看看程序哪里有问题

[复制链接]
发表于 2021-3-22 20:28:45 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
# 载入FD001数据集
file_load = 'd:/python_keras/predict aircraft engineer RUL/Dataset/PM_train_guiyihua.csv'
train_df = pd.read_csv(file_load, encoding='gbk')
# print(train_df)
new_features = ['cycle_norm', 's4', 's11', 's12', 's9', 's7', 's14', 's15',
                's21', 's2', 's3', 's20', 'setting1', 's13', 's8', 's17']
x_t = train_df[new_features]
y_t = train_df['RUL']

# 模型参数设置
# 对模型进行优化(每层神经元数量, 每层丢弃率, 迭代步数, 批次样本数)
d = [0.05, 0.1, 0.2, 0.3, 0.4, 0.5]
neuron = [1024, 512, 256, 128, 64, 32]
epochs = [10, 20, 50, 100, 1000]
batch_size = [10, 25, 50, 100, 200]
seq_length = [10, 20, 30, 50]

def model_t_c(d, n1, n2):
    model = Sequential()
    model.add(GRU(n1, input_shape=(1, 16)))
    model.add(Dropout(d))
    model.add(Dense(n2, activation='sigmoid'))
    model.add(Dense(1, activation='linear'))

    model.compile(optimizer='rmsprop', loss='mse', metrics='mae')
    model.summary()
    return model

model_t = model_t_c(0.2, 256, 256)

# 贝叶斯优化
hyperparameters = {"d": Real(0, 1, prior='log-uniform'), "n1":Real(0, 1e4, prior='uniform'), "n2":Real(0, 1e4, prior='uniform')}
Kmodel = KerasClassifier(build_fn=model_t, verbose=1)
bayesian = BayesSearchCV(estimator = Kmodel, search_spaces = hyperparameters, n_iter = 200, cv = 5, random_state=42, n_jobs = -1, scoring='neg_mean_squared_error')
bayesian.fit(x_t, y_t)
print(f'Best parameters: {bayesian.best_params_}')
print(f'Best score: {bayesian.best_score_}')


执行上述代码之后会出现下面的错误,目前没有找到解决方案
TypeError: can't pickle _thread.RLock objects
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

发表于 2021-3-24 11:14:32 | 显示全部楼层
pickle只能保存python基本类型的对象,如列表,字典。thread.RLock应该是模块中定义的对象,所以不能保存。
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-1-16 06:35

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表