(* Title: HOL/TPTP/mash_eval.ML
Author: Jasmin Blanchette, TU Muenchen
Copyright 2012
Evaluate proof suggestions from MaSh (Machine-learning for Sledgehammer).
*)
signature MASH_EVAL =
sig
type params = Sledgehammer_Provers.params
val evaluate_mash_suggestions :
Proof.context -> params -> string -> string -> unit
end;
structure MaSh_Eval : MASH_EVAL =
struct
open Sledgehammer_Fact
open Sledgehammer_MaSh
val MePoN = "MePo"
val MaShN = "MaSh"
val MeshN = "Mesh"
val IsarN = "Isar"
val all_names = map (rpair () o nickname_of) #> Symtab.make
fun evaluate_mash_suggestions ctxt params sugg_file_name out_file_name =
let
val out_path = out_file_name |> Path.explode
val _ = File.write out_path ""
fun print s = (tracing s; File.append out_path (s ^ "\n"))
val {provers, max_facts, slice, type_enc, lam_trans, timeout, ...} =
Sledgehammer_Isar.default_params ctxt []
val prover = hd provers
val slack_max_facts = generous_max_facts (the max_facts)
val sugg_path = sugg_file_name |> Path.explode
val lines = sugg_path |> File.read_lines
val css = Sledgehammer_Fact.clasimpset_rule_table_of ctxt
val facts = all_facts ctxt true false Symtab.empty [] [] css
val all_names = all_names (facts |> map snd)
val mepo_ok = Unsynchronized.ref 0
val mash_ok = Unsynchronized.ref 0
val mesh_ok = Unsynchronized.ref 0
val isar_ok = Unsynchronized.ref 0
fun with_index facts s = (find_index (curry (op =) s) facts + 1, s)
fun index_string (j, s) = s ^ "@" ^ string_of_int j
fun str_of_res label facts ({outcome, run_time, used_facts, ...}
: Sledgehammer_Provers.prover_result) =
let val facts = facts |> map (fn ((name, _), _) => name ()) in
" " ^ label ^ ": " ^
(if is_none outcome then
"Success (" ^ ATP_Util.string_from_time run_time ^ "): " ^
(used_facts |> map (with_index facts o fst)
|> sort (int_ord o pairself fst)
|> map index_string
|> space_implode " ") ^
(if length facts < the max_facts then
" (of " ^ string_of_int (length facts) ^ ")"
else
"")
else
"Failure: " ^
(facts |> take (the max_facts) |> tag_list 1
|> map index_string
|> space_implode " "))
end
fun solve_goal (j, line) =
let
val (name, suggs) = extract_query line
val th =
case find_first (fn (_, th) => nickname_of th = name) facts of
SOME (_, th) => th
| NONE => error ("No fact called \"" ^ name ^ "\".")
val goal = goal_of_thm (Proof_Context.theory_of ctxt) th
val (_, hyp_ts, concl_t) = ATP_Util.strip_subgoal ctxt goal 1
val isar_deps = isar_dependencies_of all_names th |> these
val facts = facts |> filter (fn (_, th') => thm_ord (th', th) = LESS)
val mepo_facts =
Sledgehammer_MePo.mepo_suggested_facts ctxt params prover
slack_max_facts NONE hyp_ts concl_t facts
|> Sledgehammer_MePo.weight_mepo_facts
val (mash_facts, mash_unks) =
find_mash_suggestions slack_max_facts suggs facts [] []
|>> weight_mash_facts
val mess = [(0.5, (mepo_facts, [])), (0.5, (mash_facts, mash_unks))]
val mesh_facts = mesh_facts slack_max_facts mess
val isar_facts = find_suggested_facts (map (rpair 1.0) isar_deps) facts
fun prove ok heading get facts =
let
val facts =
facts |> map get
|> Sledgehammer_Fact.maybe_instantiate_inducts ctxt hyp_ts
concl_t
|> take (the max_facts)
val res as {outcome, ...} =
run_prover_for_mash ctxt params prover facts goal
val _ = if is_none outcome then ok := !ok + 1 else ()
in str_of_res heading facts res end
val [mepo_s, mash_s, mesh_s, isar_s] =
[fn () => prove mepo_ok MePoN fst mepo_facts,
fn () => prove mash_ok MaShN fst mash_facts,
fn () => prove mesh_ok MeshN I mesh_facts,
fn () => prove isar_ok IsarN fst isar_facts]
|> map (fn f => f ())
in
["Goal " ^ string_of_int j ^ ": " ^ name, mepo_s, mash_s, mesh_s,
isar_s]
|> cat_lines |> print
end
fun total_of heading ok n =
" " ^ heading ^ ": " ^ string_of_int (!ok) ^ " (" ^
Real.fmt (StringCvt.FIX (SOME 1))
(100.0 * Real.fromInt (!ok) / Real.fromInt n) ^ "%)"
val inst_inducts = Config.get ctxt Sledgehammer_Fact.instantiate_inducts
val options =
[prover, string_of_int (the max_facts) ^ " facts",
"slice" |> not slice ? prefix "dont_", the_default "smart" type_enc,
the_default "smart" lam_trans, ATP_Util.string_from_time timeout,
"instantiate_inducts" |> not inst_inducts ? prefix "dont_"]
val n = length lines
in
print " * * *";
print ("Options: " ^ commas options);
List.app solve_goal (tag_list 1 lines);
["Successes (of " ^ string_of_int n ^ " goals)",
total_of MePoN mepo_ok n,
total_of MaShN mash_ok n,
total_of MeshN mesh_ok n,
total_of IsarN isar_ok n]
|> cat_lines |> print
end
end;