org.datavec.api.io.labels

Class ParentPathLabelGenerator

java.lang.Object

org.datavec.api.io.labels.ParentPathLabelGenerator

All Implemented Interfaces:

Serializable, PathLabelGenerator

public class ParentPathLabelGenerator
extends Object
implements PathLabelGenerator

Returns as label the base name of the parent file of the path (the directory).

Author:

saudet

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ParentPathLabelGenerator()

Method Summary

Modifier and Type	Method and Description
Writable	getLabelForPath(String path)
Writable	getLabelForPath(URI uri)
boolean	inferLabelClasses() If true: infer the set of possible label classes, and convert these to integer indexes.

Methods inherited from class java.lang.Object

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

Constructor Detail

ParentPathLabelGenerator

public ParentPathLabelGenerator()

Method Detail

getLabelForPath

public Writable getLabelForPath(String path)

Specified by:

getLabelForPath in interface PathLabelGenerator

getLabelForPath

public Writable getLabelForPath(URI uri)

Specified by:

getLabelForPath in interface PathLabelGenerator

inferLabelClasses

public boolean inferLabelClasses()

Description copied from interface: PathLabelGenerator

If true: infer the set of possible label classes, and convert these to integer indexes. If when true, the returned Writables should be text writables.

For regression use cases (or PathLabelGenerator classification instances that do their own label -> integer assignment), this should return false.

Specified by:

inferLabelClasses in interface PathLabelGenerator

Returns:

whether label classes should be inferred