Tensorflow TFlearn - 随机森林回归

Question

我试图使用TFlearn（nee skflow）TensorForestEstimator进行回归，并且当我适合（）时它会转储核心。在我的ForestHParams中，设置num_classes = 0和regression = True。这些是回归的正确设置吗？我使用的是Tensorflow 0.10.0

Answer 1

num_classes = 1是正确的设置。还尝试在适合（）的监视器kwarg中传递random_forest.TensorForestLossMonitor（或者可能只调用LossMonitor，具体取决于版本）。当树长到最大尺寸时，它将终止训练。或者，fit（）有一个你可以使用的max_steps kwargs（它不会进入ForestHParams，你只需将它传递给函数）。