bvandeusen
74fef908d2
Graduated heads can now apply their tag without a human — gated so it's safe:
- FIRING GATE: a head fires only when the master switch (head_auto_apply_enabled,
default OFF) is on AND it has >= head_auto_apply_min_positives (default 30)
clean labels. A precise-looking but under-supported low-N head can't spray tags.
- auto_apply_sweep (heads.py): streams every embedded image in chunks, scores
against the eligible heads (numpy, no sklearn), applies each head's tag where
score >= its auto_apply_threshold and the tag isn't already applied/rejected,
with source='head_auto' (distinguishable + reversible). dry_run counts only.
- HeadAutoApplyRun (migration 0059) tracks each sweep / preview; apply_head_tags
task (ml queue) + scheduled_apply_head_tags daily beat (no-op unless enabled)
+ recovery sweep + retention(20).
- API: POST /api/heads/auto-apply {dry_run} (202 / 409 running / 400 disabled),
GET /api/heads/auto-apply (recent runs + per-concept report). Settings
head_auto_apply_enabled + min_positives via /api/ml/settings.
Tests: sweep applies above threshold, dry-run writes nothing, skips under-
supported + ungraduated heads; API disabled/dry-run/conflict guards.
NEXT (slice 2): the observability the operator asked for — per-concept misfire
(auto-applied-then-removed) + under-fire tracking, time-series snapshots, and a
reporting API to tune. Slice 3: the UI (enable, preview, trends).
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa
97 lines
4.2 KiB
Python
97 lines
4.2 KiB
Python
"""MLSettings — single-row table holding ML pipeline tunables."""
|
||
|
||
from datetime import datetime
|
||
|
||
from sqlalchemy import (
|
||
Boolean,
|
||
CheckConstraint,
|
||
DateTime,
|
||
Float,
|
||
Integer,
|
||
String,
|
||
func,
|
||
)
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from .base import Base
|
||
|
||
|
||
class MLSettings(Base):
|
||
__tablename__ = "ml_settings"
|
||
# Bare name — Base.metadata's naming convention prepends ck_<table>_,
|
||
# producing the final ck_ml_settings_singleton (matches migration 0003).
|
||
__table_args__ = (CheckConstraint("id = 1", name="singleton"),)
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||
suggestion_threshold_character: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=0.70
|
||
)
|
||
# Default raised 0.50 → 0.70 on 2026-06-02 — operator-flagged 0.50
|
||
# surfaced too many low-confidence picks; 0.70 keeps the rail
|
||
# signal-rich while still surfacing more than the original 0.95
|
||
# which hid almost everything. Operator-tunable via Settings → ML.
|
||
suggestion_threshold_general: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=0.70
|
||
)
|
||
centroid_similarity_threshold: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=0.55
|
||
)
|
||
# Ingest floor: tagger predictions below this confidence are not stored
|
||
# (tagger.Tagger.infer). Default 0.70 — the suggestion path already
|
||
# filters at 0.70 and the centroid/learned path covers low-confidence
|
||
# preferred tags, so the sub-0.70 tail is redundant weight (it had
|
||
# bloated image_record's TOAST to ~100 GB; plan-task #764). Operator-
|
||
# tunable via Settings → ML; must stay ≤ the suggestion thresholds.
|
||
tagger_store_floor: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=0.70
|
||
)
|
||
min_reference_images: Mapped[int] = mapped_column(
|
||
Integer, nullable=False, default=5
|
||
)
|
||
# Video tagging (#747). Sample one frame every N seconds (fixed CADENCE, not a
|
||
# fixed count) so a tag's frame-presence reflects real screen time regardless
|
||
# of video length; cap the total so a long video can't explode into hundreds
|
||
# of inferences (the cadence stretches past the cap). A tag is kept only if it
|
||
# appears in >= video_min_tag_frames sampled frames (≈ that many × interval
|
||
# seconds on screen) — duration-independent noise rejection. Operator-tunable.
|
||
video_frame_interval_seconds: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=4.0
|
||
)
|
||
video_max_frames: Mapped[int] = mapped_column(
|
||
Integer, nullable=False, default=64
|
||
)
|
||
video_min_tag_frames: Mapped[int] = mapped_column(
|
||
Integer, nullable=False, default=3
|
||
)
|
||
# Tagging-v2 head training (#114). The head is the suggestion source that
|
||
# LEARNS from the operator's tags (replacing Camie + centroid). A concept
|
||
# needs >= head_min_positives labelled images before a head is trained;
|
||
# head_auto_apply_precision is the precision bar a head must clear (at some
|
||
# operating point) to "graduate" into earned auto-apply. Operator-tunable.
|
||
head_min_positives: Mapped[int] = mapped_column(
|
||
Integer, nullable=False, default=8
|
||
)
|
||
head_auto_apply_precision: Mapped[float] = mapped_column(
|
||
Float, nullable=False, default=0.97
|
||
)
|
||
# Earned auto-apply (#114). A graduated head fires (tags images without a
|
||
# human) ONLY when this master switch is on AND the head has at least
|
||
# head_auto_apply_min_positives clean labels — so a precise-looking but
|
||
# under-supported low-N head can't spray tags across the library. Off by
|
||
# default; the operator enables after previewing. Operator-tunable.
|
||
head_auto_apply_enabled: Mapped[bool] = mapped_column(
|
||
Boolean, nullable=False, default=False
|
||
)
|
||
head_auto_apply_min_positives: Mapped[int] = mapped_column(
|
||
Integer, nullable=False, default=30
|
||
)
|
||
tagger_model_version: Mapped[str] = mapped_column(
|
||
String(128), nullable=False, default="camie-tagger-v2"
|
||
)
|
||
embedder_model_version: Mapped[str] = mapped_column(
|
||
String(128), nullable=False, default="siglip-so400m-patch14-384"
|
||
)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), nullable=False, server_default=func.now()
|
||
)
|