updates to prospectivity_models.py

DARPA-CRITICALMAAS · Sep 25, 2024 · 887f6b3 · 887f6b3
1 parent e50be9e
commit 887f6b3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -1253,9 +1253,13 @@ classDiagram
  }
 
  class NeuralNetUserOptions {
- smoothing: float | None = 0.5
- dropout: float | None = 0.5
- negative_sampling_fraction: tuple[float, float] | None = (0.0, 0.25)
+ likely_negative_range: tuple[float, float] | None = (0.1,1.0)
+ fraction_train_split: float | None = 0.8
+ upsample_multiplier: float | None = 20.0
+ dropout: tuple[float, float, float] | None = (0.0, 0.25, 0.25)
+ learning_rate: float | None = 1e-3
+ weight_decay: float | None = 1e-2
+ smoothing: float | None = 0.3
  }
 
  class NeighborhoodDecay {

diff --git a/cdr_schemas/prospectivity_models.py b/cdr_schemas/prospectivity_models.py
@@ -40,8 +40,6 @@ class NeuralNetUserOptions(BaseModel):
  )
 
 
-
-
 class NeighborhoodFunction(str, Enum):
  GAUSSIAN = "gaussian"
  BUBBLE = "bubble"