A PyTorch neural network for predicting multiple (x, y) coordinate pairs from tabular features.
1from inference import load_model, predict
2import numpy as np
3
4model, stats, config = load_model()
5
6# Your input: 20 features
7features = np.random.randn(1, 20).astype(np.float32)
8coords = predict(model, features, stats)
9
10# coords shape: (1, 10) -> [(x0, y0), (x1, y1), ...]
11points = [(coords[0, i*2], coords[0, i*2+1]) for i in range(5)]
12print(points)
This model repository was generated by
ML Intern, an agent for machine learning research and development on the Hugging Face Hub.