p2ch14: TypeError: linspace() missing 1 required positional arguments: "steps"

[cmosguy]

So I am getting the following error on this line :

dlwpt-code/p2ch14/training.py

Line 410 in d6c0210

threshold = torch.linspace(1, 0)

TypeError: linspace() missing 1 required positional arguments: "steps"

threshold = torch.linspace(1, 0)
tpr = (metrics_t[None, METRICS_PRED_P_NDX, posLabel_mask] >= threshold[:, None]).sum(1).float() / pos_count
fpr = (metrics_t[None, METRICS_PRED_P_NDX, negLabel_mask] >= threshold[:, None]).sum(1).float() / neg_count
fp_diff = fpr[1:]-fpr[:-1]
tp_avg  = (tpr[1:]+tpr[:-1])/2

Not sure why linspace was used if the threshold was supposed to be a constant. Does anyone know what the proper fix is?

[apoorvakashi]

If you see in chapter 14 of the book, we're trying to calculate the ROC/AUC metrics. For that, we're considering multiple thresholds between 0 and 1.
So I think this might work.
threshold = torch.linspace(1,0, steps = 5)

[Rockung]

From PyTorch 1.11 linspace requires the steps argument. Use steps=100 to restore the previous behavior.