Skip to content

Commit edba972

Browse files
committed
Merge branch 'add-tsne-algorithm' of https://github.com/Khansa435/Python into add-tsne-algorithm
2 parents d1a552d + ef68a5f commit edba972

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

machine_learning/t_stochastic_neighbour_embedding.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,9 @@ def compute_pairwise_affinities(data_matrix: ndarray, sigma: float = 1.0) -> nda
4141
"""
4242
n_samples = data_matrix.shape[0]
4343
squared_sum = np.sum(np.square(data_matrix), axis=1)
44-
squared_distance = np.add(np.add(-2 * np.dot(data_matrix, data_matrix.T), squared_sum).T, squared_sum)
44+
squared_distance = np.add(
45+
np.add(-2 * np.dot(data_matrix, data_matrix.T), squared_sum).T, squared_sum
46+
)
4547

4648
affinity_matrix = np.exp(-squared_distance / (2 * sigma**2))
4749
np.fill_diagonal(affinity_matrix, 0)
@@ -125,7 +127,10 @@ def apply_tsne(
125127

126128
gradient = 4 * (
127129
np.dot((affinity_diff * numerator_matrix), embedding)
128-
- np.multiply(np.sum(affinity_diff * numerator_matrix, axis=1)[:, np.newaxis], embedding)
130+
- np.multiply(
131+
np.sum(affinity_diff * numerator_matrix, axis=1)[:, np.newaxis],
132+
embedding,
133+
)
129134
)
130135

131136
embedding_increment = momentum * embedding_increment - learning_rate * gradient
@@ -168,4 +173,3 @@ def main() -> None:
168173
if __name__ == "__main__":
169174
doctest.testmod()
170175
main()
171-

0 commit comments

Comments
 (0)