katz/su.ml at master · alexy/katz · GitHub

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
open Common
open Soc_run_local
open Getopt

let byMass'    = ref true
let minDays'   = ref 7
let minCap'    = ref 1e-35
let jumpProb'  = ref 0.2
let attStrat'  = ref UniformAttachment
let mark'      = ref ""

let specs =
[
  ('c',"mincap",None,Some (fun x -> minCap' := float_of_string x));
  ('k',"mark",  None,Some (fun x -> mark' := x));
  ('u',"byusers",(set byMass' false),None);
  (noshort,"nomindays",(set minDays' 0),None);
  ('d',"mindays",None,Some (fun n -> minDays' := int_of_string n));
  ('j',"jump",None,Some (fun x -> jumpProb' := float_of_string x));
  (noshort,"auni",(set attStrat' UniformAttachment),None);
  (noshort,"amen",(set attStrat' MentionsAttachment),None)
]

let () =
  let args = getOptArgs specs in

  let byMass,   minDays,   minCap,   jumpProb,   attStrat,   mark =
      !byMass', !minDays', !minCap', !jumpProb', !attStrat', !mark' in

  let (dstartsName,drnumsName,saveBase,dreps',day') =
  match args with
  | dstartsName::drnumsName::saveBase::dreps'::day'::[] ->
      (dstartsName,drnumsName,saveBase,Some dreps',Some day')
  | dstartsName::drnumsName::saveBase::[] ->
      (dstartsName,drnumsName,saveBase,None,None)
  | _ -> failwith "usage: sg dtartsName drnumsName saveBase [drepsName initDay]"
  in
  let saveSuffix = saveBase^".mlb" in
  let drepsName  = "dreps-"^saveSuffix in
  let dmentsName = "dments-"^saveSuffix in
  let capsName   = "caps-"^saveSuffix in
  let skewName   = "skew-"^saveSuffix in
  let jumpName   = "jump-"^saveSuffix in
  leprintfln "reading dstarts from %s and drnums from %s, saving dreps in %s, dments in %s, dcaps in %s and dskews in %s"
    dstartsName drnumsName drepsName dmentsName capsName skewName;

  let strategyName = showStrategy attStrat in
  leprintfln "options: byMass=%b, minDays=%d, minCap=%e, jumpProb=%f, strategy=%s"
    byMass minDays minCap jumpProb strategyName;

  let dstarts: starts      = loadData dstartsName in
  let tLoadDStarts =  Some "-- loaded dstarts timing: " |> getTiming in
  let drnums: day_rep_nums = loadData drnumsName in
  let tLoadDRnums  =  Some "-- loaded denums timing: "  |> getTiming in

  let (initDrepsO,initDayO) = match dreps' with
  | Some drepsName ->
          let day = day' |> Option.get |> int_of_string in
          leprintfln "based on %s through day %d" drepsName day;
          let dreps: graph = loadData drepsName in
          (Some dreps, Some day)
  | _ -> (None, None) in

  let opts = {optSocRun with (* maxDaysSR= maxDays; *) byMassSR= byMass;
                             initDrepsSR= initDrepsO; initDaySR= initDayO;
                             minCapSR= minCap;
 (* minCapDaysSR=0 means raw 1 capital for attachment, no maturity at all! *)
                             minCapDaysSR= minDays;
                             jumpProbSR= jumpProb; attachmentStrategySR= attStrat}
                             in

  let {drepsSG =dreps; dmentsSG =dments},
    dcaps,dskews,countsTimings = socRun dstarts drnums opts in

  let edgeCounts,tSocRun = L.split countsTimings in

  leprintfln "computed sgraph, now saving dreps in %s, dments in %s, dcaps in %s, dskews in %s, jumps in %s"
    drepsName dmentsName capsName skewName jumpName;
  saveData dreps  drepsName;
  let tSavingDReps  =  Some "-- saved dreps timing: "  |> getTiming in
  saveData dments dmentsName;
  let tSavingDMents =  Some "-- saved dments timing: " |> getTiming in
  saveData dcaps  capsName;
  let tSavingDCaps  =  Some "-- saved dcaps timing: "  |> getTiming in
  saveData dskews skewName;
  let tSavingDSkews =  Some "-- saved dskews timing: " |> getTiming in

  saveData edgeCounts jumpName;

  let ts = List.rev (tSavingDSkews::tSavingDCaps::tSavingDMents::tSavingDReps::tSocRun@[tLoadDRnums;tLoadDStarts]) in
  printf "timings: %s\n" (show_float_list ts);