什么导致我的随机林代码中出现这种奇怪的TypeError?

时间:2017-07-09 07:20:22

标签: python csv machine-learning tensorflow random-forest

我有一些非常简单的代码训练来自csv的数据的随机森林。代码,减去导入和常量可以在下面找到:

TypeError: Input 'input_data' of 'CountExtremelyRandomStats' Op has type float64 that does not match expected type of float32.

当我到达est.fit线时虽然崩溃说了以下内容:

apply_op
    (prefix, dtypes.as_dtype(input_arg.type).name))

显然,这是在以下代码行中名为op_def_library.py的一些tensorflow文件中出现的:

{{1}}

不确定是什么导致这种情况。我似乎在说从csv读取的值应该是float32类型。真的很沮丧。关于如何解决它的任何想法?

1 个答案:

答案 0 :(得分:0)

我遇到了同样的错误,tf.cast()解决了这个问题。

training_set = tf.cast(training_set, tf.float32)

看到这个答案: TypeError when training Tensorflow Random Forest using TensorForestEstimator