-
Notifications
You must be signed in to change notification settings - Fork 0
/
gibbs.hpp
31 lines (25 loc) · 805 Bytes
/
gibbs.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
// Copyright (c) 2021 Christopher Taylor
//
// SPDX-License-Identifier: BSL-1.0
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
#ifndef __GIBBS_HPP__
#include <vector>
#include <cmath>
#include <cstdint>
#include <blaze/Math.h>
#include "drand.hpp"
using blaze::DynamicMatrix;
using blaze::DynamicVector;
using blaze::CompressedMatrix;
void gibbs(
CompressedMatrix<double> const& dwcm,
DynamicMatrix<double> & tdcm,
DynamicMatrix<double> & twcm,
std::vector<std::size_t> & tokens, DynamicVector<double> & ztot,
DynamicVector<double> & probs,
drand & dr,
const std::size_t n_topics, const double N, const double alpha, const double beta);
#endif