net.sourceforge.cilib.nn.domain.LambdaGammaSolutionConversionStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cilib-library Show documentation

A library of composable components enabling simpler Computational Intelligence

There is a newer version: 0.8

/**           __  __
 *    _____ _/ /_/ /_    Computational Intelligence Library (CIlib)
 *   / ___/ / / / __ \   (c) CIRG @ UP
 *  / /__/ / / / /_/ /   http://cilib.net
 *  \___/_/_/_/_.___/
 */
package net.sourceforge.cilib.nn.domain;

import net.sourceforge.cilib.nn.NeuralNetwork;
import net.sourceforge.cilib.nn.NeuralNetworks;
import net.sourceforge.cilib.nn.architecture.visitors.ArchitectureVisitor;
import net.sourceforge.cilib.nn.architecture.visitors.LambdaGammaVisitor;
import net.sourceforge.cilib.type.types.Type;
import net.sourceforge.cilib.type.types.container.Vector;

public class LambdaGammaSolutionConversionStrategy implements SolutionConversionStrategy {

    private int activationFuncCount = 0;
    private int weightCount = 0;

    @Override
    public SolutionConversionStrategy initialise(NeuralNetwork neuralNetwork) {
        activationFuncCount = NeuralNetworks.countActivationFunctions(neuralNetwork);
        weightCount = NeuralNetworks.countWeights(neuralNetwork);
        return this;
    }

    @Override
    public ArchitectureVisitor interpretSolution(Type solution) {
        return new LambdaGammaVisitor((Vector)solution, weightCount, activationFuncCount);
    }

    protected int getActivationFuncCount() {
        return activationFuncCount;
    }

    protected int getWeightCount() {
        return weightCount;
    }
}