org.deeplearning4j.datasets.iterator.callbacks.DataSetDeserializer Maven / Gradle / Ivy

Go to download

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

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.datasets.iterator.callbacks;

import lombok.extern.slf4j.Slf4j;
import org.nd4j.linalg.dataset.DataSet;

import java.io.File;

/**
 * This callback does DataSet deserialization of a given file.
 *
 * @author [email protected]
 */
@Slf4j
public class DataSetDeserializer implements FileCallback {
    @Override
    public  T call(File file) {
        DataSet dataSet = new DataSet();
        dataSet.load(file);
        return (T) dataSet;
    }
}