homebrew-core/Formula/vowpal-wabbit.rb

95 lines
3.3 KiB
Ruby

class VowpalWabbit < Formula
desc "Online learning algorithm"
homepage "https://github.com/VowpalWabbit/vowpal_wabbit"
# pull from git tag to get submodules
url "https://github.com/VowpalWabbit/vowpal_wabbit.git",
tag: "8.10.1",
revision: "3887696a19c34baaff5ebd0dd7c68a7437cd0175"
license "BSD-3-Clause"
head "https://github.com/VowpalWabbit/vowpal_wabbit.git"
bottle do
sha256 cellar: :any, arm64_big_sur: "5edbdd1e6e78252daacb66822d856c7c4268a426b7e076b3acb579fe2b8b59d1"
sha256 cellar: :any, big_sur: "8f7d47864a841bd3d247813f9c0bc62d034bb34e0bd1bc5a79680d0b2bb5decf"
sha256 cellar: :any, catalina: "7c0d4169175b8cede06d8b2896c27b4c5cd9f0a13d50a355d8bd9ae372afcfe4"
sha256 cellar: :any, mojave: "0f7007ba2de3d476c4e77f6e1949cd11fe4dc50d8fba99d01550b1cbea610092"
end
depends_on "cmake" => :build
depends_on "flatbuffers" => :build
depends_on "rapidjson" => :build
depends_on "spdlog" => :build
depends_on "boost"
depends_on "fmt"
depends_on "zlib"
def install
ENV.cxx11
# The project provides a Makefile, but it is a basic wrapper around cmake
# that does not accept *std_cmake_args.
# The following should be equivalent, while supporting Homebrew's standard args.
mkdir "build" do
system "cmake", "..", *std_cmake_args,
"-DBUILD_TESTS=OFF",
"-DRAPIDJSON_SYS_DEP=ON",
"-DFMT_SYS_DEP=ON",
"-DSPDLOG_SYS_DEP=ON",
"-DBUILD_FLATBUFFERS=ON"
system "make", "install"
end
bin.install Dir["utl/*"]
rm bin/"active_interactor.py"
rm bin/"new_version"
rm bin/"vw-validate.html"
rm bin/"clang-format"
rm_r bin/"flatbuffer"
end
test do
(testpath/"house_dataset").write <<~EOS
0 | price:.23 sqft:.25 age:.05 2006
1 2 'second_house | price:.18 sqft:.15 age:.35 1976
0 1 0.5 'third_house | price:.53 sqft:.32 age:.87 1924
EOS
system bin/"vw", "house_dataset", "-l", "10", "-c", "--passes", "25", "--holdout_off",
"--audit", "-f", "house.model", "--nn", "5"
system bin/"vw", "-t", "-i", "house.model", "-d", "house_dataset", "-p", "house.predict"
(testpath/"csoaa.dat").write <<~EOS
1:1.0 a1_expect_1| a
2:1.0 b1_expect_2| b
3:1.0 c1_expect_3| c
1:2.0 2:1.0 ab1_expect_2| a b
2:1.0 3:3.0 bc1_expect_2| b c
1:3.0 3:1.0 ac1_expect_3| a c
2:3.0 d1_expect_2| d
EOS
system bin/"vw", "--csoaa", "3", "csoaa.dat", "-f", "csoaa.model"
system bin/"vw", "-t", "-i", "csoaa.model", "-d", "csoaa.dat", "-p", "csoaa.predict"
(testpath/"ect.dat").write <<~EOS
1 ex1| a
2 ex2| a b
3 ex3| c d e
2 ex4| b a
1 ex5| f g
EOS
system bin/"vw", "--ect", "3", "-d", "ect.dat", "-f", "ect.model"
system bin/"vw", "-t", "-i", "ect.model", "-d", "ect.dat", "-p", "ect.predict"
(testpath/"train.dat").write <<~EOS
1:2:0.4 | a c
3:0.5:0.2 | b d
4:1.2:0.5 | a b c
2:1:0.3 | b c
3:1.5:0.7 | a d
EOS
(testpath/"test.dat").write <<~EOS
1:2 3:5 4:1:0.6 | a c d
1:0.5 2:1:0.4 3:2 4:1.5 | c d
EOS
system bin/"vw", "-d", "train.dat", "--cb", "4", "-f", "cb.model"
system bin/"vw", "-t", "-i", "cb.model", "-d", "test.dat", "-p", "cb.predict"
end
end