tensorflow:加载csv数据文件并训练mod

用户

我不熟悉tensorflow。我需要加载数据集来训练我的模型。我的数据集样本看起来像

TRAINING_FILE.iloc[0:5,0:5]

    num_var_1   num_var_2   num_var_3   num_var_4   num_var_5
0   -0.010655   0.040182    0.0     1.800000e-07    -0.011319
1   -0.006542   0.157872    0.0     2.105000e-06    -0.010966
2   -0.010626   0.089140    0.0     3.550000e-07    -0.011286
3   -0.010626   0.227239    0.0     1.050000e-06    -0.011159
4   -0.008947   0.160410    0.0     2.105000e-06    -0.010966

我用tensorflow文档中提到的代码加载这个csv文件

^{pr2}$

当我编译脚本时,我得到了以下错误

Traceback (most recent call last): File "train.py", line 31, in features_dtype = np.float32) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/datasets/base.py", line 48, in load_csv_with_header n_samples = int(header[0]) ValueError: invalid literal for int() with base 10: '-0.0106550312'


已被浏览了4890次
更新日期: 2020-10-28 19:05:20
1 个回答
fefe Tyson

这些看起来都像float,但是load_csv_with_header正在寻找一个dtype为target_dtype(在您的例子中是整数)的标签列。您可以使用target_column参数选择此列,但默认情况下它是最后一个。在

因此,您要么需要将label数据类型切换为float(如果您要预测实际值),要么向数据添加一个label列。在

评论 - 2020年9月11日 16:34

最新Python问答

推荐Python问答