【发布时间】:2019-02-21 14:42:29
【问题描述】:
我们有什么方法可以使用内置的tensorflow 优化器来针对特定指标进行优化?如果没有,如何实现?例如。如果我只想专注于最大化我的分类器的 F 分数,是否可以在 tensorflow 中这样做?
estimator = tf.estimator.LinearClassifier(
feature_columns=feature_cols,
config=my_checkpointing_config,
model_dir=output_dir,
optimizer=lambda: tf.train.FtrlOptimizer(
learning_rate=tf.train.exponential_decay(
learning_rate=0.1,
global_step=tf.train.get_or_create_global_step(),
decay_steps=1000,
decay_rate=0.96)))
我正在尝试在获得更好的 F 分数的基础上专门优化我的分类器。尽管使用了衰减的learning_rate 和 300 个训练步骤,但我得到的结果不一致。在检查日志中的指标时,我发现precision、recall 和accuracy 的行为非常不稳定。尽管增加了训练步骤的数量,但没有显着改善。所以我想如果我能让优化器更专注于提高整体的 F 分数,我可能会得到更好的结果。因此问题。有什么我遗漏的吗?
【问题讨论】:
标签: python tensorflow machine-learning