2016-07-06 10 views
2

Ich habe meine Tensorflow von 0,7 bis 0,9 auf Python3 aktualisiert. Und jetzt kann ich nicht meine vorherigen gespeicherten Modelle mit skflow (tensorflow.contrib.learn) wiederherstellen. Hier ist die Beispielcode Beispiel, das am Tensorflow 0.7 gearbeitet wurde.tensorflow 0.9 skflow Modell speichern und wiederherstellen funktioniert nicht

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

Auf Tensorflow 0.9 habe ich diese Fehler erhalten.

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

Antwort

0

I Belive speichern und wiederherstellen wurden zugunsten des model_dir param veraltet, wenn die Schätzer/Regressor Aufbau:

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

nur zu klären, wenn ich/fit Modellbahn in xyz.py und Verwendung Vorhersage des Codes in mno.py (ohne Trainingsmodell); wird es funktionieren ? – turtle