python - 如何使用 Google Colab 从 tensorflow fully_connected 获取权重

标签 python tensorflow google-colaboratory

我正在尝试在训练后从模型中提取权重。这是一个代码。 1-如何将模型保存在之前上传的目录路径中? 2.如何获得权重? 3. 我有一些用于建模但没有类的 csv 文件。我如何使用此模型估计类别?

# Preprocess
import pandas as pd
import keras
from sklearn.model_selection import train_test_split
# Prepare data
churn = pd.read_csv('Churn_Modelling3.csv')
# split data into train and test sets
x_train, x_test, y_train, y_test = train_test_split(churn.iloc[:,0:10],churn.iloc[:, 10], test_size=0.2)                    
# Create a Model
import tensorflow as tf
from keras.models import Sequential
from keras.layers import Dense
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D
from tensorflow.keras.layers import MaxPool2D
from tensorflow.keras.layers import Flatten
from tensorflow.keras.layers import Dense
model = Sequential()
# Input layer
model.add(Dense(32, activation=tf.nn.sigmoid,input_dim=x_train.shape[1]))
# Hidden layers
model.add(Dense(64, activation=tf.nn.sigmoid))
model.add(Dense(32, activation=tf.nn.sigmoid))
# Output layer
model.add(Dense(1, activation=tf.nn.sigmoid))
# Compile the Model
from keras.optimizers import SGD
sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile( optimizer= 'sgd', loss='binary_crossentropy', metrics=['accuracy'])
# Fit the model
history = model.fit(x_train, y_train, epochs=20, batch_size=128,validation_data=(x_test, y_test))
# ٍEvaluate the Model
score = model.evaluate(x_test, y_test, batch_size=128)

最佳答案

  1. 模型如何保存到之前上传的目录路径? => 您可以使用以下任何命令,

     `model.save_weights('./Weights/Weights')` or 
     `tf.keras.experimental.export_saved_model` or 
      `model.save`
    
  2. 如何获取权重? => 我们可以使用以下任何命令加载权重:

    model.load_weights('my_model_weights.h5') or         
    keras.models.load_model('my_model.h5') or
    
    tf.keras.experimental.load_from_saved_model
    
  3. 我有一些用于建模但没有类的 csv 文件。我怎样才能用这个模型估计类(class)? => 其中一种方法是为 CSV 数据标签的每个实例迭代以下代码。

    ynew = model.predict(Xnew)
    

更多信息,您可以引用以下链接:

https://www.tensorflow.org/tutorials/keras/save_and_restore_models

https://www.tensorflow.org/guide/saved_model

https://www.tensorflow.org/api_docs/python/tf/keras/Model

关于python - 如何使用 Google Colab 从 tensorflow fully_connected 获取权重,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57807716/

相关文章:

python - 概念 - 查询数据库和页面提供有限的属性

python - 使用 pandas 从 CSV 加载随机样本

python - 如何计算梯度来欺骗图像分类器?

python - 无法从 flow_from_dataframe 进行训练 意外没有。类(class)数

tensorflow - VGG 的每一层有多少个神经元?

python - 自动将数据帧的列标题传递到一列中,并将相应的值作为新列

python - 为什么我的 scikit 学习 HashingVectorizo​​r 给我 float binary = True 集?

python - 在复杂模型上使用 Pytorch 进行修剪

python - 设置 TensorBoard 以在 Google Colab 中为 TensorFlow 对象检测模型运行 eval.py 作业

github - Git 文件夹到 google colab