def test_fit(self): from rsatoolbox.rdm import RDMs rdm = np.random.rand(2, 6) pattern_descriptors = {'test': ['a', 'b', 'c', 'd']} rdm_descriptors = {'ind': np.array([1, 2])} rdm_obj = RDMs(rdm, dissimilarity_measure='euclid', pattern_descriptors=pattern_descriptors, rdm_descriptors=rdm_descriptors) m = model.ModelWeighted('Test Model', rdm_obj) train = rdm_obj.subset('ind', 2) theta = m.fit(train)
def test_fit(self): from rsatoolbox.rdm import RDMs rdm = np.random.rand(5, 15) pattern_descriptors = {'test': ['a', 'b', 'c', 'd', 'e', 'f']} rdm_descriptors = {'ind': np.array([1, 2, 3, 1, 2])} rdm_obj = RDMs(rdm, dissimilarity_measure='euclid', pattern_descriptors=pattern_descriptors, rdm_descriptors=rdm_descriptors) m = model.ModelInterpolate('Test Model', rdm_obj) train = rdm_obj.subset('ind', 2) theta = m.fit(train) _ = m.predict(theta)