首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何将数组形式的model.get_weight保存到csv中?

如何将数组形式的model.get_weight保存到csv中?
EN

Stack Overflow用户
提问于 2019-06-18 19:56:46
回答 2查看 1.9K关注 0票数 0

我是一个初学者,开始学习在Keras后端Tensorflow中编写代码。我使用的是python 2.7

我有一个穿着眼镜的模特,训练结束后,我想检查一下我的体重。

已编辑

# fix random seed for reproducibility (split training and validation set)
seed = 7
np.random.seed(seed)

# load data
(X_train, y_train), (X_test, y_test) = mnist.load_data()

# reshape to be [samples][pixels][width][height]
X_train = X_train.reshape(X_train.shape[0], 1, 28, 28).astype('float32')
X_test = X_test.reshape(X_test.shape[0], 1, 28, 28).astype('float32')

# normalize inputs from 0-255 to 0-1
X_train = X_train / 255
X_test = X_test / 255

# one hot encode outputs (label encoding)
y_train = np_utils.to_categorical(y_train)
y_test = np_utils.to_categorical(y_test)
num_classes = y_test.shape[1]

def tempsigmoid(x, temp=0.5):
    return K.sigmoid(x/temp)


def baseline_model():
    # create model
    model = Sequential()
    model.add(Conv2D(32, (5, 5), input_shape=(1, 28, 28), activation='relu'))
    #model.add(Dense, input_shape = (1,28,28), Activation='relu')
    model.add(MaxPooling2D(pool_size=(2, 2)))
    model.add(Dropout(0.2))
    model.add(Flatten())
    model.add(Dense(100, activation='relu'))
    model.add(Dense(num_classes, activation=tempsigmoid))

    # Compile model
    model.compile(loss='mae', optimizer=SGD(lr=0.1), metrics=['accuracy'])
    return model

    # build the model
model = baseline_model()

earlystopper = EarlyStopping(monitor='val_loss', min_delta=0.1, patience=0, verbose=2, mode='auto')

# Fit the model
model.fit(X_train, y_train, validation_data=(X_test, y_test), epochs=5, batch_size=200, verbose=2, callbacks=[earlystopper])

# Final evaluation of the model
scores = model.evaluate(X_test, y_test, verbose=0)
print("CNN Error: %.2f%%" % (100-scores[1]*100))
# print("Metrics(Test loss & Test Accuracy): ")
print(scores)

weight = model.get_weights()
print(weight)

我得到数组中的权重,如图所示。如何将权重数组保存为csv文件?

weight array

我尝试使用model.save_weight(),我有一个h5格式的输出文件,但是当我想用numpy打开它时,它只显示其中的一小部分。我想当我可以保存为csv格式时,我将得到数据的完整显示。

我曾尝试过使用numpy python将h5转换为csv,如图所示。

# to save weight after output
model.save_weights('Result/w_output.h5')

Tried to display full array with numpy

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2019-06-19 09:38:47

通过使用下面的代码,我成功地将数组权重保存为csv

weight = model.get_weights()
np.savetxt('weight.csv' , weight , fmt='%s', delimiter=',')
票数 1
EN

Stack Overflow用户

发布于 2019-06-18 20:24:25

我假设您希望每个数字列表都在它自己的行中。如果是这种情况,则需要将权重调整为以下形状: weight = [data1,data2,data3...,data11,data12,data13......]。functions for reforming list

然后,您只需将其写入/附加到csv文件。您可以使用pandas或csv库。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/56648591

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档