训练数据集中的函数mnist.train.next_batch()有什么用?

时间:2019-01-02 06:46:02

标签: python tensorflow

我正在使用TensorFlow使用胶囊网络训练自己的数据集。在训练mnist数据集时,它包含函数mnist.train.next_batch(批量大小)。如何使用TensorFlow替换此功能以训练自己的数据集?

1 个答案:

答案 0 :(得分:0)

函数样本<script> $(document).ready(function(){ $("#submit").click(function(e){ e.preventDefault(); product_name = $("#product_name").val(); color = $("#colorWell").val(); var formData = new FormData(); $.each($("#product_image"), function (i, obj) { $.each(obj.files, function (j, file) { formData.append('product_image[' + i + ']', file); }); }); formData.append('product_name', product_name); formData.append('color', color); $.ajax({ type:"POST", data:formData, processData: false, contentType: false, url:"<?php echo base_url(); ?>admin/products", success:function(data){ alert(data); } }); }); }); </script> 中的数据来自经过改组的训练数据集,然后返回批次进行训练。

您可以编写自己的batch_size方法来执行相同的操作,也可以根据需要对其进行修改。然后在训练模型时类似地使用它。