41 lines
1.4 KiB
Ruby
41 lines
1.4 KiB
Ruby
class Lightgbm < Formula
|
|
desc "Fast, distributed, high performance gradient boosting framework"
|
|
homepage "https://github.com/microsoft/LightGBM"
|
|
url "https://github.com/microsoft/LightGBM/archive/v2.3.1.tar.gz"
|
|
sha256 "b2b8ed5e06be21444c4e9defd66877510d1800a650543f75bfc1b28f77745b84"
|
|
|
|
bottle do
|
|
cellar :any
|
|
sha256 "b8b078e8458186a058a47ba94d302dfb3688cae329112594bbbb125712c5c79a" => :catalina
|
|
sha256 "eefdb9c1c9749755431ac7fafc09bc9d3d31d73d6d9077f4e0a9d6d89ae4d032" => :mojave
|
|
sha256 "afaa412201c008a88980e9ecd9967d4ccfacdaa967229f084457e3780cba3b78" => :high_sierra
|
|
end
|
|
|
|
depends_on "cmake" => :build
|
|
depends_on "libomp"
|
|
|
|
def install
|
|
mkdir "build" do
|
|
libomp = Formula["libomp"]
|
|
args = std_cmake_args
|
|
args << "-DOpenMP_C_FLAGS=\"-Xpreprocessor -fopenmp -I#{libomp.opt_include}\""
|
|
args << "-DOpenMP_C_LIB_NAMES=omp"
|
|
args << "-DOpenMP_CXX_FLAGS=\"-Xpreprocessor -fopenmp -I#{libomp.opt_include}\""
|
|
args << "-DOpenMP_CXX_LIB_NAMES=omp"
|
|
args << "-DOpenMP_omp_LIBRARY=#{libomp.opt_lib}/libomp.dylib"
|
|
args << "-DAPPLE_OUTPUT_DYLIB=ON"
|
|
|
|
system "cmake", *args, ".."
|
|
system "make"
|
|
system "make", "install"
|
|
end
|
|
pkgshare.install "examples"
|
|
end
|
|
|
|
test do
|
|
cp_r (pkgshare/"examples/regression"), testpath
|
|
cd "regression" do
|
|
system "#{bin}/lightgbm", "config=train.conf"
|
|
end
|
|
end
|
|
end
|