StreamingFileFromFeatures.h

浏览该文件的文档.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2011 Shashwat Lal Das
 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
 */
#ifndef __STREAMING_FILEFROMFEATURES_H__
#define __STREAMING_FILEFROMFEATURES_H__

#include <shogun/lib/config.h>

#include <shogun/io/streaming/StreamingFile.h>
#include <shogun/io/SGIO.h>

namespace shogun
{
class CFeatures;

class CStreamingFileFromFeatures: public CStreamingFile
{
public:
    CStreamingFileFromFeatures();

    CStreamingFileFromFeatures(CFeatures* feat);

    CStreamingFileFromFeatures(CFeatures* feat, float64_t* lab);

    virtual ~CStreamingFileFromFeatures();

    virtual void set_features(CFeatures* feat)
    {
        ASSERT(feat)
        features=feat;
    }

    virtual void set_labels(float64_t* lab)
    {
        ASSERT(lab)
        labels=lab;
    }

    virtual const char* get_name() const
    {

        return "StreamingFileFromFeatures";

    }

protected:

    CFeatures* features;

    float64_t* labels;
};
}
#endif //__STREAMING_FILEFROMFEATURES_H__