Skip to content

Commit a599a02

Browse files
authored
fixed r2score - pred and true values in wrong order (#84)
1 parent fab132a commit a599a02

File tree

3 files changed

+3
-7
lines changed

3 files changed

+3
-7
lines changed

util/metrics.q

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ rmsle:{rmse . log(x;y)+1}
3737
mae:{avg abs x-y}
3838
mape:{100*avg abs 1-x%y}
3939
smape:{100*avg abs[y-x]%abs[x]+abs y}
40-
r2score:{1-sse[x;y]%sse[x]avg x}
40+
r2score:{1-sse[y;x]%sse[y]avg y}
4141

4242
/ t-score for a test (one sample)
4343
tscore:{[x;mu](avg[x]-mu)%sdev[x]%sqrt count x}

util/tests/metric.t

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,9 +138,9 @@ plaintabn:plaintab,'([]x4:1 3 0n)
138138
.ml.smape[xm;ym]~{smape[x;y]}'[flip xm;flip ym]
139139
.ml.smape[x;x]~0f
140140
.ml.smape[1 0n 4 2 0n;1 2 4 3 1]~6.666666666666666667
141-
.ml.r2score[xf;yf] ~ r2[xf;yf]`
141+
.ml.r2score[xf;yf] ~ r2[yf;xf]`
142142
.ml.r2score[xf;xf] ~ r2[xf;xf]`
143-
.ml.r2score[1 2 3;2 2 2] ~ r2[1 2 3;2 2 2]`
143+
.ml.r2score[2 2 2;1 2 3] ~ r2[1 2 3;2 2 2]`
144144
.ml.r2score[x;x]~1f
145145
.ml.r2score[1 0n 4 2 0n;1 2 4 2 1]~1f
146146
.ml.tscore[x;y] ~first stats[`:ttest_1samp][x;y]`

util/tests/mlpy.p

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,3 @@ p)def< mean_absolute_percentage_error(y_true, y_pred):
77

88
p)def< smape(A, F):
99
return 100 * np.mean(np.abs(F - A) / (np.abs(A) + np.abs(F)))
10-
11-
p)def< f2_score(y_true, y_pred):
12-
y_true, y_pred, = np.array(y_true), np.array(y_pred)
13-
return fbeta_score(y_true, y_pred, beta=2, average='samples')

0 commit comments

Comments
 (0)