xt2dnn.cpp

decltype(auto) convertDataset(const xt::xarray<float>& X, const xt::xarray<float>& Y)
{
	size_t numData = X.shape()[0];
	std::vector<vec_t> x(numData);
	std::vector<label_t> y(numData);
	for(size_t i=0;i<numData;i++)
	{
		auto xb = xt::view(X, i, xt::all());
		x[i] = vec_t(xb.begin(), xb.end());
		
		auto yb = xt::view(Y, i, xt::all());
		y[i] = xt::argmax(yb)[0];
	}
	
	//shuffle dataset
	shuffle(x.begin(), x.end(), std::mt19937(42));
	shuffle(y.begin(), y.end(), std::mt19937(42));
	
	return std::tuple(x, y);
}

auto [x, y] = convertDataset(X, Y);