我试图在 OCaml 中使用逻辑回归。我需要将它用作我正在解决的另一个问题的黑盒。我找到了以下网站:
http://math.umons.ac.be/anum/en/software/OCaml/Logistic_Regression/
我将此站点中的以下代码(进行了一些修改 - 我定义了自己的 iris_features 和 iris_label)粘贴到名为logistic_regression.ml 的文件中:
open Scanf
open Format
open Bigarray
open Lacaml.D
let log_reg ?(lambda=0.1) x y =
(* [f_df] returns the value of the function to maximize and store
its gradient in [g]. *)
let f_df w g =
let s = ref 0. in
ignore(copy ~y:g w); (* g ← w *)
scal (-. lambda) g; (* g = -λ w *)
for i = 0 to Array.length x - 1 do
let yi = float y.(i) in
let e = exp(-. yi *. dot w x.(i)) in
s := !s +. log1p e;
axpy g ~alpha:(yi *. e /. (1. +. e)) ~x:x.(i);
done;
-. !s -. 0.5 *. lambda *. dot w w
in
let w = Vec.make0 (Vec.dim x.(0)) in
ignore(Lbfgs.F.max f_df w);
w
let iris_features = [1 ; 2 ; 3] ;;
let iris_labels = 2 ;;
let proba w x y = 1. /. (1. +. exp(-. float y *. dot w x))
let () =
let sol = log_reg iris_features iris_labels in
printf "w = %a\n" Lacaml.Io.pp_fvec sol;
let nwrongs = ref 0 in
for i = 0 to Array.length iris_features - 1 do
let p = proba sol iris_features.(i) iris_labels.(i) in
printf "Label = %i prob = %g => %s\n" iris_labels.(i) p
(if p > 0.5 then "correct" else (incr nwrongs; "wrong"))
done;
printf "Number of wrong labels: %i\n" !nwrongs
我有以下问题:
- 在尝试编译代码时,我收到错误消息:“
Error: Unbound module Lacaml
”。我已经安装了 Lacaml;做了几次 opam init,试图提供一个标志 -package = Lacaml ;我不知道如何解决这个问题? - 如您所见,我已经定义了自己的 iris_features 和 iris_labels 版本——类型是否正确,即函数 log_reg 中的类型是 x int 列表,而 y 的类型是 int?