org.deeplearning4j.zoo.ModelMetaData Maven / Gradle / Ivy
package org.deeplearning4j.zoo;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* Metadata describing a model, including input shapes. This is helpful for instantiating
* the model programmatically and ensuring appropriate inputs are used.
*
* @deprecated As of May 10, 2018. Will be removed in v1.1. Getters are now available on the inputShape from the ZooModel implementation.
*/
@Getter
@AllArgsConstructor
@Deprecated
public class ModelMetaData {
private int[][] inputShape;
private int numOutputs;
private ZooType zooType;
/**
* If number of inputs are greater than 1, this states that the
* implementation should use MultiDataSet.
* @return
*/
public boolean useMDS() {
return inputShape.length > 1 ? true : false;
}
}