55 lines
1.5 KiB
Nix
55 lines
1.5 KiB
Nix
{ lib, stdenv, fetchFromGitHub, fetchpatch, cmake, boost, eigen, rapidjson, spdlog, zlib }:
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "vowpal-wabbit";
|
|
version = "9.6.0";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "VowpalWabbit";
|
|
repo = "vowpal_wabbit";
|
|
rev = version;
|
|
sha256 = "sha256-iSsxpeTRZjIhZaYBeoKLHl9j1aBIXWjONmAInmKvU/I=";
|
|
};
|
|
|
|
patches = [
|
|
# Fix x86_64-linux build by adding missing include
|
|
# https://github.com/VowpalWabbit/vowpal_wabbit/pull/4275
|
|
(fetchpatch {
|
|
url = "https://github.com/VowpalWabbit/vowpal_wabbit/commit/0cb410dfc885ca1ecafd1f8a962b481574fb3b82.patch";
|
|
sha256 = "sha256-bX3eJ+vMTEMAo3EiESQTDryBP0h2GtnMa/Fz0rTeaNY=";
|
|
})
|
|
];
|
|
|
|
nativeBuildInputs = [ cmake ];
|
|
|
|
buildInputs = [
|
|
boost
|
|
eigen
|
|
rapidjson
|
|
spdlog
|
|
zlib
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DVW_INSTALL=ON"
|
|
"-DBUILD_JAVA=OFF"
|
|
"-DBUILD_PYTHON=OFF"
|
|
"-DRAPIDJSON_SYS_DEP=ON"
|
|
"-DFMT_SYS_DEP=ON"
|
|
"-DSPDLOG_SYS_DEP=ON"
|
|
"-DVW_BOOST_MATH_SYS_DEP=ON"
|
|
"-DVW_EIGEN_SYS_DEP=ON"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "Machine learning system focused on online reinforcement learning";
|
|
homepage = "https://github.com/VowpalWabbit/vowpal_wabbit/";
|
|
license = licenses.bsd3;
|
|
longDescription = ''
|
|
Machine learning system which pushes the frontier of machine learning with techniques such as online,
|
|
hashing, allreduce, reductions, learning2search, active, and interactive and reinforcement learning
|
|
'';
|
|
maintainers = with maintainers; [ jackgerrits ];
|
|
platforms = platforms.unix;
|
|
};
|
|
}
|