2016-07-06 15 views
2

私はpython3.Andでテンソルフローを0.7から0.9に更新しましたが、skflow(tensorflow.contrib.learn)を使用して前回保存したモデルを復元することはできません。テンソルフロー0.7で作業されたサンプルコードの例tensorflow 0.9 skflow model保存と復元が機能しません

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/') 

私はこのエラーを受け取りました。

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

答えて

0

私は推定/回帰を構築するときmodel_dirのparamの賛成で廃止されました保存と復元信じて:

regressor = skflow.TensorFlowLinearRegressor(model_dir='/home/model/') 
regressor.fit(X, boston.target) 
... 
estimator = skflow.TensorFlowLinearRegressor(model_dir='/home/model/') 
estimator.predict(...) 
+0

だけ明確にするために、私はxyz.pyおよび使用中/フィットモデルを訓練する場合mno.pyのコードを予測する(訓練モデルなしで)。それは働くでしょうか? – turtle

関連する問題