2

Python3 でテンソルフローを 0.7 から 0.9 に更新しました。以前に保存したモデルを skflow(tensorflow.contrib.learn) で復元できません。テンソルフロー 0.7 で動作したサンプル コード例を次に示します。

import tensorflow.contrib.learn as skflow
from sklearn import datasets, metrics, preprocessing

boston = datasets.load_boston()
X = preprocessing.StandardScaler().fit_transform(boston.data)
regressor = skflow.TensorFlowLinearRegressor()
regressor.fit(X, boston.target)
score = metrics.mean_squared_error(regressor.predict(X), boston.target)
print ("MSE: %f" % score)

regressor.save('/home/model/')

classifier = skflow.TensorFlowEstimator.restore('/home/model/')

tensorflow 0.9 で、このエラーを受け取りました。

AttributeError: 'TensorFlowLinearRegressor' object has no attribute '_restore'
4

1 に答える 1

0

model_dir私は、estimator/regressor を構築するときに paramを支持して save と restore が非推奨になったと信じています:

regressor = skflow.TensorFlowLinearRegressor(model_dir='/home/model/')
regressor.fit(X, boston.target)
...
estimator = skflow.TensorFlowLinearRegressor(model_dir='/home/model/')
estimator.predict(...)
于 2016-08-19T16:11:43.710 に答える