The disclosure notably relates to a computer-implemented method of machine-learning. The method includes obtaining a dataset including 3D modeled objects which each represent a respective mechanical part. The dataset has one or more sub-datasets. Each sub-dataset forms at least a part of the dataset. The method further includes, for each respective sub-dataset, determining a base template and learning a neural network configured for inference of deformations of the base template each into a respective 3D modeled object. The base template is a 3D modeled object which represents a centroid of the 3D modeled objects of the sub-dataset. The learning includes a training based on the sub-dataset. This constitutes an improved method of machine-learning with a dataset including 3D modeled objects which each represent a respective mechanical part.