Parameter Constrained Neural Network

← Back to index

Model overview

Parameter constrained neural network classifier using PyTorch with configurable hidden layers to predict taxonomy IDs from product tags. Each taxonomy path is encoded as a p-adic integer and the model minimizes p-adic distance.

• Total train samples: 30,760
• Total test samples: 7,690

Key hyperparameters

• Prime base(s): 79
• Hidden layer sizes: 27
• Max tags used: 32

Cross-validation results