dataset.h

来自「简单分类器 VC++ 用于简单的分类」· C头文件 代码 · 共 42 行

//类CDataSet: 数据集类
//职责: 
//  1. 管理多项矢量数据
//  2. 对属性值进行离散化处理
//  3. 提取某一类别的数据?

#pragma once

#include "dataitem.h"
#include <vector>
using namespace std;

//typedef pair<int, CDataItem*> ClassDataPair;
//typedef map<int, CDataItem*>::iterator DataSetInterator;

class CDataSet
{
public:
	CDataSet(void);
	CDataSet(DWORD nSize, int nDimensions=0);
	~CDataSet(void);

protected:
	vector<CDataItem*> m_DataItems; //仍然用vector保存多项数据
	int m_nDimension; //各矢量数据的维数
	int m_nClasses;

public:
	bool Load(CString& sDataFile); //从数据文件加载
	bool Save(CString& sDataFile); //保存数据到文件
	bool Add(CDataItem *pItem); //添加一个数据项
	void Clear();
	DWORD Count()
	{	return (DWORD)m_DataItems.size();	}
	int Dimension() //返回数据的维数
	{	return m_nDimension;	}
	CDataItem& operator [] (DWORD nIndex); //重载[]操作符, 以存取各项数据.
	int Classes()
	{	return m_nClasses;	}
	void SetSize(DWORD nSize, int nDimension=0);
};