fix bug in fil.go (#50)

getumen · Aug 2, 2024 · e5763e7 · e5763e7
1 parent 6a33a6b
commit e5763e7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/go/fil.go b/go/fil.go
@@ -152,7 +152,7 @@ func (m *FILModel) PredictSingleClassScore(
 
 	result := make([]float32, numRow)
 	for i := 0; i < numRow; i++ {
-		result[i] = resultRaw[i*m.NumClass()+1]
+		result[i] = resultRaw[i*2+1]
 	}
 	return result, nil
 }
@@ -165,6 +165,3 @@ func (m *FILModel) Close() error {
 	return err
 }
 
-func (m *FILModel) NumClass() int {
-	return m.raw.NumClass()
-}