代码之家  ›  专栏  ›  技术社区  ›  netskink

如何使用张力板确定重要特征?

  •  0
  • netskink  · 技术社区  · 5 年前

    我有一个二进制分类器模型,它是使用估计器api构建的。这个模型有大约300个特征和一个标签。我用的是一个有175个神经元的单隐层模型。

    我用5000个训练步骤来运行模型。在分成训练、测试和验证集之前,我有278个一个类的实例和178个第二个类的实例。它并不完全平衡。

    也就是说,我得到的精度是0.984。

    所以,我试图解释直方图的结果,看看是否可以用它来确定哪些特征比其他特征更重要。下面是直方图选项卡中的一些图片。

    hidden_layer_0 activation

    以下是logits的截图。

    logits activation

    0 回复  |  直到 5 年前
        1
  •  1
  •   Travis Kaufman    5 年前

    根据这些截图,这些直方图看起来像是来自隐藏层和logits的激活值。我不相信这会给你任何关于你的特性是如何影响模型输出的指示,因为 这些直方图只是计算这些图形操作的输出值的分布 .

    监测 而不是 分析 make it easier to understand, debug, and optimize TensorFlow programs .

    Tensorflow Model Analysis ( TFMA公司

    Gif of example TFMA use as seen in its README

    getting started guide 可能是一个很好的起点!