2016-06-27 11 views
6

मैं कुछ पायथन-ब्लैकबॉक्स फ़ंक्शन द्वारा गणना की गई प्रति-रन जानकारी टेंसरबोर्ड में रिकॉर्ड करना चाहता हूं।टेंसरबोर्ड लॉगिंग गैर-टेंसर (numpy) जानकारी (एयूसी)

विशेष रूप से, मैं sess.run() चलाने के बाद sklearn.metrics.auc का उपयोग करके कल्पना कर रहा हूं।

यदि "auc" वास्तव में एक टेंसर नोड था, तो जीवन सरल होगा। हालांकि, सेटअप अधिक की तरह है:

stuff=sess.run() 
auc=auc(stuff) 

यदि वहाँ ऐसा करने का एक और अधिक tensorflow-onic तरह से मुझे लगता है कि में दिलचस्पी है। मेरे वर्तमान सेटअप में अलग-अलग ट्रेन & परीक्षण ग्राफ बनाना शामिल है।

यदि ऊपर बताए गए कार्य को पूरा करने का कोई तरीका है, तो मुझे इसमें रूचि भी है।

उत्तर

10

आप इस कोड का उपयोग अपने स्वयं के डेटा के साथ एक कस्टम सारांश बना सकते हैं:

tf.Summary(value=[tf.Summary.Value(tag="auc", simple_value=auc)]))

तो फिर तुम सारांश लेखक अपने आप से है कि सारांश जोड़ सकते हैं। (step जोड़ने के लिए मत भूलना)।

+0

वास्तव में यह काम करता है। – user3391229

+5

@ डैनमैन - उत्तर के लिए धन्यवाद, लेकिन आपका मतलब क्या है, "एक कदम जोड़ने के लिए मत भूलना"? – RobR