diff options
-rw-r--r-- | README.md | 2 | ||||
-rw-r--r-- | ot/da.py | 2 |
2 files changed, 3 insertions, 1 deletions
@@ -68,6 +68,8 @@ cd cudamat python setup.py install --user # for user install (no root) ``` +obviously you need CUDA installed and a compatible GPU. + ## Examples The examples folder contain several examples and use case for the library. The full documentation is available on [Readthedocs](http://pot.readthedocs.io/) @@ -620,7 +620,7 @@ class OTDA(object): self.wt=wt self.M=dist(xs,xt,metric=self.metric) - self.normalize() + self.normalizeM(norm) self.G=emd(ws,wt,self.M) self.computed=True |