FactorAnalysis.cpp

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2013 Fernando J. Iglesias Garcia
 * Copyright (C) 2011-2013 Fernando J. Iglesias Garcia
 */

#include <shogun/converter/FactorAnalysis.h>
#ifdef HAVE_EIGEN3
#include <shogun/lib/tapkee/tapkee_shogun.hpp>
#include <shogun/features/DenseFeatures.h>

using namespace shogun;

CFactorAnalysis::CFactorAnalysis() :
        CEmbeddingConverter()
{
    // Sentinel value, it will be set appropriately if not modified by set_max_iteration
    m_max_iteration = 0;
    m_epsilon = 1e-5;
    init();
}

void CFactorAnalysis::init()
{
    SG_ADD(&m_max_iteration, "max_iteration", "maximum number of iterations", MS_NOT_AVAILABLE);
    SG_ADD(&m_epsilon, "epsilon", "convergence parameter", MS_NOT_AVAILABLE);
}

CFactorAnalysis::~CFactorAnalysis()
{
}

const char* CFactorAnalysis::get_name() const
{
    return "FactorAnalysis";
}

void CFactorAnalysis::set_max_iteration(const int32_t max_iteration)
{
    m_max_iteration = max_iteration;
}

int32_t CFactorAnalysis::get_max_iteration() const
{
    return m_max_iteration;
}

void CFactorAnalysis::set_epsilon(const float64_t epsilon)
{
    m_epsilon = epsilon;
}

float64_t CFactorAnalysis::get_epsilon() const
{
    return m_epsilon;
}

CFeatures* CFactorAnalysis::apply(CFeatures* features)
{
    TAPKEE_PARAMETERS_FOR_SHOGUN parameters;
    parameters.max_iteration = m_max_iteration;
    parameters.features = (CDotFeatures*)features;
    parameters.fa_epsilon = m_epsilon;
    parameters.method = SHOGUN_FACTOR_ANALYSIS;
    parameters.target_dimension = m_target_dim;
    CDenseFeatures<float64_t>* embedding = tapkee_embed(parameters);
    return embedding;
}

#endif /* HAVE_EIGEN */