Source code for statsmodels.multivariate.factor_rotation._analytic_rotation

# -*- coding: utf-8 -*-
"""
This file contains analytic implementations of rotation methods.
"""

import numpy as np
import scipy as sp


[docs]def target_rotation(A, H, full_rank=False): r""" Analytically performs orthogonal rotations towards a target matrix, i.e., we minimize: .. math:: \phi(L) =\frac{1}{2}\|AT-H\|^2. where :math:`T` is an orthogonal matrix. This problem is also known as an orthogonal Procrustes problem. Under the assumption that :math:`A^*H` has full rank, the analytical solution :math:`T` is given by: .. math:: T = (A^*HH^*A)^{-\frac{1}{2}}A^*H, see Green (1952). In other cases the solution is given by :math:`T = UV`, where :math:`U` and :math:`V` result from the singular value decomposition of :math:`A^*H`: .. math:: A^*H = U\Sigma V, see Schonemann (1966). Parameters ---------- A : numpy matrix (default None) non rotated factors H : numpy matrix target matrix full_rank : bool (default FAlse) if set to true full rank is assumed Returns ------- The matrix :math:`T`. References ---------- [1] Green (1952, Psychometrika) - The orthogonal approximation of an oblique structure in factor analysis [2] Schonemann (1966) - A generalized solution of the orthogonal procrustes problem [3] Gower, Dijksterhuis (2004) - Procrustes problems """ ATH = A.T.dot(H) if full_rank or np.linalg.matrix_rank(ATH) == A.shape[1]: T = sp.linalg.fractional_matrix_power(ATH.dot(ATH.T), -1/2).dot(ATH) else: U, D, V = np.linalg.svd(ATH, full_matrices=False) T = U.dot(V) return T
[docs]def procrustes(A, H): r""" Analytically solves the following Procrustes problem: .. math:: \phi(L) =\frac{1}{2}\|AT-H\|^2. (With no further conditions on :math:`H`) Under the assumption that :math:`A^*H` has full rank, the analytical solution :math:`T` is given by: .. math:: T = (A^*HH^*A)^{-\frac{1}{2}}A^*H, see Navarra, Simoncini (2010). Parameters ---------- A : numpy matrix non rotated factors H : numpy matrix target matrix full_rank : bool (default False) if set to true full rank is assumed Returns ------- The matrix :math:`T`. References ---------- [1] Navarra, Simoncini (2010) - A guide to empirical orthogonal functions for climate data analysis """ return np.linalg.inv(A.T.dot(A)).dot(A.T).dot(H)
[docs]def promax(A, k=2): r""" Performs promax rotation of the matrix :math:`A`. This method was not very clear to me from the literature, this implementation is as I understand it should work. Promax rotation is performed in the following steps: * Determine varimax rotated patterns :math:`V`. * Construct a rotation target matrix :math:`|V_{ij}|^k/V_{ij}` * Perform procrustes rotation towards the target to obtain T * Determine the patterns First, varimax rotation a target matrix :math:`H` is determined with orthogonal varimax rotation. Then, oblique target rotation is performed towards the target. Parameters ---------- A : numpy matrix non rotated factors k : float parameter, should be positive References ---------- [1] Browne (2001) - An overview of analytic rotation in exploratory factor analysis [2] Navarra, Simoncini (2010) - A guide to empirical orthogonal functions for climate data analysis """ assert k > 0 # define rotation target using varimax rotation: from ._wrappers import rotate_factors V, T = rotate_factors(A, 'varimax') H = np.abs(V)**k/V # solve procrustes problem S = procrustes(A, H) # np.linalg.inv(A.T.dot(A)).dot(A.T).dot(H); # normalize d = np.sqrt(np.diag(np.linalg.inv(S.T.dot(S)))) D = np.diag(d) T = np.linalg.inv(S.dot(D)).T return A.dot(T), T