this is what i got:
5 data points:
Epoch: 1/100
training r2: 0.13436292977865993, mean loss: 0.33448025584220886
…
Epoch: 100/100
training r2: 0.1343612927439954, mean loss: 0.33448025584220886
2 data points:
Epoch: 1/100
training r2: 1.0, mean loss: 0.6390723586082458
Epoch: 100/100
training r2: 1.0, mean loss: 0.6390724778175354
1 data point:
ValueError: x and y must have length at least 2.