Skip to content

Commit cc39e3b

Browse files
AlexanderKalistratovZzEeKkAa
authored andcommitted
fix pca_numba_n
1 parent a08706b commit cc39e3b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpbench/benchmarks/pca/pca_numba_n.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def pca(data, dims_rescaled_data=2):
3939
evectors = evectors[:, :dims_rescaled_data]
4040

4141
# carry out the transformation on the data using eigenvectors
42-
tdata = np.dot(evectors.T, data.T).T
42+
tdata = np.dot(evectors.T.astype(data.dtype), data.T).T
4343

4444
# return the transformed data, eigenvalues, and eigenvectors
4545
return tdata, evalues, evectors

0 commit comments

Comments
 (0)