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: 24,724
• Total test samples: 6,181

Key hyperparameters

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

Cross-validation results