2016-02-17 50 views
9

Tôi đang cố gắng để tạo ra một tập hợp của nhiều mô hình được đào tạo. Tất cả các mô hình có cùng một biểu đồ và chỉ khác nhau theo trọng số của nó. Tôi đang tạo biểu đồ mô hình bằng cách sử dụng tf.get_variable. Tôi có một số điểm kiểm tra khác nhau (với các trọng số khác nhau) cho cùng một kiến ​​trúc đồ thị và tôi muốn tạo một mô hình mẫu cho mỗi trạm kiểm soát.Làm thế nào để tạo ra toàn bộ trong tensorflow?

Làm cách nào tôi có thể tải nhiều điểm kiểm tra mà không ghi đè trọng số đã tải trước đó?

Khi tôi tạo biểu đồ của mình với tf.get_variable, cách duy nhất tôi có thể tạo nhiều biểu đồ là bằng cách chuyển đối số reuse = True. Bây giờ nếu tôi thử thay đổi tên của các biến đồ thị của tôi kèm theo phương thức xây dựng trong một phạm vi mới (để chúng trở thành không thể chia sẻ với các đồ thị được tạo khác) trước khi tải, thì điều này sẽ không hoạt động vì tên mới sẽ khác với trọng lượng và tôi sẽ không thể tải nó.

+0

Tôi chưa thử, nhưng dưới đây là một số mã tham chiếu: https://github.com/eske/seq2seq/blob/master/translate/__main__.py#L190 Tóm lại, tác giả tạo ra nhiều phiên làm nhiều điểm kiểm tra và trong mỗi phiên, anh ta khôi phục điểm kiểm tra tương ứng. – tnq177

+0

Một số mã tham chiếu có liên quan hơn: https://github.com/Hvass-Labs/TensorFlow-Tutorials/blob/master/05_Ensemble_Learning.ipynb – Deepank

+0

@cesarsalgado: Tôi có cùng một vấn đề. Tôi đang sử dụng inception-v4 trong tf-slim. Bạn đã giải quyết nó như thế nào? – Jame

Trả lời

-1

Tôi cũng đã trả lời câu hỏi này trên Reddit nhưng chỉ trong trường hợp những người khác gặp phải sự cố này, tôi đang đăng ký lại ở đây và thêm các giải thích rõ ràng hơn.

Nếu bạn sử dụng Scikit Flow, bạn sẽ có thể dễ dàng lưu và khôi phục các mô hình, đưa ra các dự đoán khác nhau từ các mô hình khác nhau và sau đó xây dựng toàn bộ các dự đoán đó. Hãy xem cách bạn có thể lưu và khôi phục chúng từ trang ví dụ của chúng tôi.

Ngoài ra còn có predict_proba() bạn có thể sử dụng, intead của predict(), để có được xác suất dự đoán. Nó sẽ được dễ dàng sau đó để xây dựng trung bình đơn giản, xếp chồng, vv, các loại khác nhau của ensembles.

Hy vọng điều này sẽ hữu ích.

Các vấn đề liên quan