布局模型训练方法、装置以及设备
摘要:
本公开提供了一种布局模型训练方法、装置以及设备,涉及人工智能技术领域,具体涉及工业、数据处理、深度学习等技术领域。该方法的一具体实施方式包括:获取样本部件集合和样本布局图,其中,样本部件集合包括需要布局到样本布局图中的样本部件,且可拼装成样本产品;提取样本部件集合的样本特征张量和样本布局图的样本布局矩阵;利用样本部件集合的样本特征张量和样本布局图的样本布局矩阵对原始模型进行训练,得到布局模型,其中,布局模型用于从部件集合中选取下一个布局到布局图中的部件。该实施方式以部件特征和布局图状态感知为输入,训练出能够从部件集合中选取下一个布局到布局图的部件的布局模型,使得布局模型具有出色的布局能力。
0/0