org.deeplearning4j.zoo

Class ModelSelector

java.lang.Object

org.deeplearning4j.zoo.ModelSelector

public class ModelSelector
extends Object

Helper class for selecting multiple models from the zoo.

Author:

Justin Long (crockpotveggies)

Constructor Summary

Constructors

Constructor and Description
ModelSelector()

Method Summary

Modifier and Type	Method and Description
static Map<ZooType,ZooModel>	select(int numLabels, int seed, int iterations, WorkspaceMode workspaceMode, ZooType... zooTypes) Select specific models from the zoo.
static Map<ZooType,ZooModel>	select(WorkspaceMode workspaceMode, ZooType... zooTypes)
static Map<ZooType,ZooModel>	select(ZooType... zooTypes)
static Map<ZooType,ZooModel>	select(ZooType zooType)
static Map<ZooType,ZooModel>	select(ZooType zooType, int numLabels)
static Map<ZooType,ZooModel>	select(ZooType zooType, int numLabels, int seed, int iterations)
static Map<ZooType,ZooModel>	select(ZooType zooType, int numLabels, int seed, int iterations, WorkspaceMode workspaceMode) Select multiple models from the zoo according to type.
static Map<ZooType,ZooModel>	select(ZooType zooType, int numLabels, WorkspaceMode workspaceMode)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ModelSelector

public ModelSelector()

Method Detail

select

public static Map<ZooType,ZooModel> select(ZooType zooType)

select

public static Map<ZooType,ZooModel> select(ZooType zooType,
                                           int numLabels)

select

public static Map<ZooType,ZooModel> select(ZooType zooType,
                                           int numLabels,
                                           WorkspaceMode workspaceMode)

select

public static Map<ZooType,ZooModel> select(ZooType zooType,
                                           int numLabels,
                                           int seed,
                                           int iterations)

select

public static Map<ZooType,ZooModel> select(ZooType zooType,
                                           int numLabels,
                                           int seed,
                                           int iterations,
                                           WorkspaceMode workspaceMode)

Select multiple models from the zoo according to type.

Parameters:

zooType -

numLabels -

seed -

iterations -

Returns:

A hashmap of zoo types and models.

select

public static Map<ZooType,ZooModel> select(WorkspaceMode workspaceMode,
                                           ZooType... zooTypes)

select

public static Map<ZooType,ZooModel> select(ZooType... zooTypes)

select

public static Map<ZooType,ZooModel> select(int numLabels,
                                           int seed,
                                           int iterations,
                                           WorkspaceMode workspaceMode,
                                           ZooType... zooTypes)

Select specific models from the zoo.

Parameters:

numLabels -

seed -

iterations -

workspaceMode -

zooTypes -

Returns:

A hashmap of zoo types and models.