来自 “ ITPUB博客 ” 链接://viewspace-758953/,如需转載请注明出处,否则将追究法律责任
打印最后5行CSV文件信息了解文件的格式,其中“survived”为标簽列
正如你看到的那样CSV 文件的每列都会有一个列名。dataset 的构造函数会自动识别这些列名如果你使用的文件的第一行不包含列名,那么需偠将列名通过字符串列表传给 make_csv_dataset 函数的 column_names 参数
CSV 数据中的有些列是分类的列。也就是说这些列只能在有限的集合中取值。
这将是后续构建模型时处理输入數据的一部分
写一个函数标准化这些值,然后将这些值改造成 2 维的张量
这里使用标准化的方法需要提前知道每列的均值。如果需要计算连续的数据流的标准化的值可以使用
将这两个特征列的集合相加并且传给 tf.keras.layers.DenseFeatures 从而创建一个进行预处理的输入层。
现在可以实例化和训练模型
当模型训练完成的时候,你可以在测试集 test_data 上检查准确性
版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。
点击添加站长微信