本发明公开了一种基于机器学习的工业无线边缘网关优化布局方案，具体步骤为：获取车间内的产线和传感器位置布局，并通过传感器数据采样器获取不同传感器在不同时刻需要上传的数据量；通过蒙特卡洛随机模拟计算传感器传输数据时消耗的总能量的期望；根据采样器得到的上传数据和蒙特卡罗随机模拟计算的能量期望，训练一个神经网络用来预测任意一组无线边缘网关位置对应的能量期望；采用启发式算法搜索无线边缘网关位置，并基于训练的神经网络嵌入启发式算法以加速搜索过程。本发明主要解决物联网工厂车间内的无线边缘网关的部署问题，明显提高了工厂车间内传感器的数据传输能效，提升了工厂的长期平均收益。