<jupyter_start><jupyter_text>Testing optimal transport<jupyter_code>import numpy as np
import pylab as pl
import ot
n = 1000
d = 2
sigma = .1

# source samples
angles = np.random.rand(n, 1) * 2 * np.pi
xs = np.concatenate((np.sin(angles), np.cos(angles)),
                    axis=1) + sigma * np.random.randn(n, 2)
xs[:n // 2, 1] += 2


# target samples
anglet = np.random.rand(n, 1) * 2 * np.pi
xt = np.concatenate((np.sin(anglet), np.cos(anglet)),
                    axis=1) + sigma * np.random.randn(n, 2)
xt[:n // 2, 1] += 2


A = np.array([[1.5, .7], [.7, 1.5]])
b = np.array([[4, 2]])
xt = xt.dot(A) + b
pl.figure(1, (5, 5))
pl.plot(xs[:, 0], xs[:, 1], '+')
pl.plot(xt[:, 0], xt[:, 1], 'o')
Ae, be = ot.da.OT_mapping_linear(xs, xt)

xst = xs.dot(Ae) + be
pl.figure(1, (5, 5))
pl.clf()
pl.plot(xs[:, 0], xs[:, 1], '+')
pl.plot(xt[:, 0], xt[:, 1], 'o')
pl.plot(xst[:, 0], xst[:, 1], '+')

pl.show()
xs.shape
be<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.24947145877378435, 'fraction_numerical': 0.046511627906976744, 'mean_word_length': 3.2466367713004485, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '232958', 'license': 'MIT', 'n_tokens_mistral': 470, 'n_tokens_neox': 437, 'n_words': 104, 'path': 'development/test_ot.ipynb', 'repo_name': 'wbishopJanelia/probabilistic_model_synthesis', 'size': '946'}