Сохранение и восстановление модели tenorflow 0.9 skflow не работает

Question

Я обновил свой тензор потока с 0,7 до 0,9 на python3.И теперь я не могу восстановить свои предыдущие сохраненные модели с помощью skflow(tenorflow.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/')

На tenorflow 0.9 я получил эти ошибки.

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

Answer 1

Я верю, что спасение и восстановление устарели в пользу model_dir Параметр при построении оценщика / регрессора:

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