Package ai.djl.huggingface.translator

Class PretrainedConfig

java.lang.Object

ai.djl.huggingface.translator.PretrainedConfig

public class PretrainedConfig
extends Object

A class representing HuggingFace config.json file.

Constructor Summary

Constructors

Constructor	Description
PretrainedConfig()

Method Summary

Modifier and Type	Method	Description
void	setId2label(Map<String,String> id2label)	Sets the id to label map.
void	setProblemType(String problemType)	Sets the problem type.

Methods inherited from class java.lang.Object

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

Constructor Details

PretrainedConfig

public PretrainedConfig()

Method Details

setProblemType

public void setProblemType(String problemType)

Sets the problem type.

Parameters:

problemType - the problem type

setId2label

public void setId2label(Map<String,String> id2label)

Sets the id to label map.

Parameters:

id2label - the id to label map