   [& {:keys [fail-if-not-exists? file]
       :or {fail-if-not-exists? true
            file (:name (model-config))}}]
-  (cl/read-model file :fail-if-not-exists? fail-if-not-exists?))
+  (let [model (cl/read-model file :fail-if-not-exists? fail-if-not-exists?)
+        context (:context model)
+        {:keys [set-context-fn]} (model-config)]
+    (if (and set-context-fn context)
+      (assoc model :context (set-context-fn context))
+      model)))
 
 (defn prime-model
   "Prime a trained or unpersisted ([[read-model]]) model for classification

   (create-instances features-set nil))
  ([features-set context]
   (log/infof "generating instances from %d feature sets" (count features-set))
   (log/tracef "feature sets: <<%s>>" (pr-str features-set))
   (let [{:keys [name set-context-fn]} (model-config)]
     (if (and set-context-fn context)
       (set-context-fn context))
     (weka/instances
      (format "%s-classify" name)
      features-set

  (:require [zensols.tabres.display-results :as dr])
  (:require [zensols.actioncli.resource :refer (resource-path)]
            [zensols.util.spreadsheet :as ss]
            [zensols.util.string :as zs]
            [zensols.model.classifier :as cl]
            [zensols.model.weka :as weka]))

