如何在tf.contrib.learn Estimator

时间:2017-02-24 17:30:10

标签: python tensorflow

在训练和预测中 tensorflow.contrib.learn 中使用 estimator.Estimator 时, modeldir 中存在以下文件:

  • 检查点
  • events.out.tfevents.1487956647
  • events.out.tfevents.1487957016
  • graph.pbtxt
  • model.ckpt-101.data-00000-的-00001
  • model.ckpt-101.index
  • model.ckpt-101.meta

当图形复杂或变量数量很大时,graph.pbtxt文件和事件文件可能非常大。这是一种不写这些文件的方法吗?由于模型重新加载只需要删除检查点文件,因此不会影响评估和预测。

3 个答案:

答案 0 :(得分:1)

如果您不想写event.out.tfevents文件。在您的代码中找到类似这些的东西并删除它们。

tfFileWriter = tf.summary.FileWriter(os.getcwd())
tfFileWriter.add_graph(sess.graph)
tfFileWriter.close()

答案 1 :(得分:1)

Google's Machine Learning Crash Course上,他们使用以下方法:

You can use the activityIndicator for showing the spinner. For that you have 
to make below changes.


1. import the activity indicator

    import {
      ActivityIndicator,
      Text,
      View,
    } from 'react-native'
    
2. set a state named isLoading in the constructor
    
    constructor(){
      super();
      this.state={
        isLoading: false
      }
    }
    
3. add activity indicator inside the render function
    
    render() {
      return (
        <View >
          {this.state.loading && <ActivityIndicator size="large" color="#0000ff" />}
        </View>
      )
    }

 

when you want to show the spinner set the value of isLoading true
 
    this.setState({ isLoading: true });

答案 2 :(得分:0)

我遇到了同样的问题,但是在事件文件不断增长的同时,找不到任何解决方案。我的理解是该文件存储了由tensorflow生成的事件。我继续将其手动删除。 有趣的是,当我运行火车序列时,当其他文件更新时,它再也没有创建。