Question

我正在尝试保存protobuf，pbtext，tf事件文件和模型文件（在Keras中为h5文件）。前三个已保存，但最后一个未发生。它抛出错误“ AttributeError：“ SVM”对象没有属性“ get_config””

如何解决它并将模型保存为linear.h5 keras格式？

我曾尝试按照stackoverflow线程的建议将scikit学习版升级到0.19.1版，但无效。

import numpy as np
import tensorflow as tf
from tensorflow import keras

X = np.zeros([157, 128])
Y = np.zeros([157], dtype=np.int32)
example_id = np.array(['%d' % i for i in range(len(Y))])
x_column_name = 'x'
example_id_column_name = 'example_id'
with tf.Graph().as_default():
    with tf.Session() train_input_fn=tf.estimator.inputs.numpy_input_fn(x=x_column_name: X, example_id_column_name: example_id},y=Y,num_epochs=None,shuffle=True)
        svm = tf.contrib.learn.SVM(example_id_column=example_id_column_name,feature_ 
        columns=(tf.contrib.layers.real_valued_column(column_name=x_column_name, dimension=128),),l2_regularization=0.1)
        svm=svm.fit(input_fn=train_input_fn, steps=10)
        print("Hi")
        file_writer = tf.summary.FileWriter('res1', sess.graph)
        tf.train.write_graph(sess.graph,'resm','file.pbtxt')
        tf.train.write_graph(sess.graph,'resm','savemodel.pb', as_text=False)
        print("Model saving in Keras")

        keras_file="linear.h5"
        keras.models.save_model(svm, keras_file)

预期---以h5格式保存的模型。

错误

AttributeError: 'SVM' object has no attribute 'get_config'

将模型另存为Keras linear.h5模型

0 个答案: