|
130 | 130 | "\n", |
131 | 131 | "def squared_euclidean_dist(u, v, axis=-1):\n", |
132 | 132 | " \"\"\"Returns squared Euclidean distance between two vectors.\"\"\"\n", |
133 | | - " return squared_euclidean_norm(u -v)\n", |
| 133 | + " return squared_euclidean_norm(u-v, axis)\n", |
134 | 134 | "\n", |
135 | 135 | "def euclidean_dist(u, v, axis=-1):\n", |
136 | 136 | " \"\"\"Return Euclidean distacne between two vectors.\"\"\"\n", |
|
164 | 164 | "language": "python", |
165 | 165 | "metadata": {}, |
166 | 166 | "outputs": [], |
167 | | - "prompt_number": 6 |
| 167 | + "prompt_number": 12 |
168 | 168 | }, |
169 | 169 | { |
170 | 170 | "cell_type": "code", |
171 | 171 | "collapsed": false, |
172 | 172 | "input": [ |
173 | 173 | "M = np.array([[1,2,3],[4,5,6]])\n", |
174 | 174 | "\n", |
175 | | - "dist = euclidean_dist\n", |
| 175 | + "# dist = euclidean_dist\n", |
| 176 | + "dist = cosine_dist\n", |
176 | 177 | "\n", |
177 | 178 | "print loop_row_pdist(M, dist), '\\n'\n", |
178 | 179 | "print broadcast_row_pdist(M, dist), '\\n'\n", |
|
186 | 187 | "output_type": "stream", |
187 | 188 | "stream": "stdout", |
188 | 189 | "text": [ |
189 | | - "[[ 0.000 5.196]\n", |
190 | | - " [ 5.196 0.000]] \n", |
| 190 | + "[[ 0.000 0.025]\n", |
| 191 | + " [ 0.025 0.000]] \n", |
191 | 192 | "\n", |
192 | | - "[[ 0.000 5.196]\n", |
193 | | - " [ 5.196 0.000]] \n", |
| 193 | + "[[ 0.000 0.025]\n", |
| 194 | + " [ 0.025 0.000]] \n", |
194 | 195 | "\n", |
195 | | - "[[ 0.000 1.414 2.828]\n", |
196 | | - " [ 1.414 0.000 1.414]\n", |
197 | | - " [ 2.828 1.414 0.000]] \n", |
| 196 | + "[[ 0.000 0.009 0.024]\n", |
| 197 | + " [ 0.009 -0.000 0.003]\n", |
| 198 | + " [ 0.024 0.003 0.000]] \n", |
198 | 199 | "\n", |
199 | | - "[[ 2.236 1.414 2.236]\n", |
200 | | - " [ 2.236 1.414 2.236]]\n" |
| 200 | + "[[ 0.000 0.009 0.024]\n", |
| 201 | + " [ 0.009 -0.000 0.003]\n", |
| 202 | + " [ 0.024 0.003 0.000]]\n" |
201 | 203 | ] |
202 | 204 | } |
203 | 205 | ], |
204 | | - "prompt_number": 7 |
| 206 | + "prompt_number": 13 |
205 | 207 | }, |
206 | 208 | { |
207 | 209 | "cell_type": "markdown", |
|
0 commit comments