org.deeplearning4j.graph.vertexfactory.StringVertexFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-graph Show documentation

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.graph.vertexfactory;

import org.deeplearning4j.graph.api.Vertex;

public class StringVertexFactory implements VertexFactory {

    private final String format;

    public StringVertexFactory(){
        this(null);
    }

    public StringVertexFactory(String format){
        this.format = format;
    }

    @Override
    public Vertex create(int vertexIdx) {
        if(format != null) return new Vertex<>(vertexIdx,String.format(format,vertexIdx));
        else return new Vertex<>(vertexIdx,String.valueOf(vertexIdx));
    }
}