55640456ff
* Add recall * Add SensitivitySpecificity metrics * Review comments * Add missing method from math * Fix pytest * Update config dict creation * Remove sentence fragment * Review comments * Add special casing for min/max
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""
|
|
segment_sum
|
|
top_k
|
|
"""
|
|
|
|
from keras_core import backend
|
|
from keras_core.backend import any_symbolic_tensors
|
|
from keras_core.operations.operation import Operation
|
|
|
|
|
|
class SegmentSum(Operation):
|
|
def call(self, data, segment_ids, num_segments=None, sorted=False):
|
|
return backend.math.segment_sum(data, segment_ids, num_segments, sorted)
|
|
|
|
|
|
def segment_sum(data, segment_ids, num_segments=None, sorted=False):
|
|
if any_symbolic_tensors((data,)):
|
|
return SegmentSum().symbolic_call(
|
|
data, segment_ids, num_segments, sorted
|
|
)
|
|
return backend.math.segment_sum(data, segment_ids, num_segments, sorted)
|
|
|
|
|
|
class TopK(Operation):
|
|
def call(self, x, k, sorted=True):
|
|
return backend.math.top_k(x, k, sorted)
|
|
|
|
|
|
def top_k(x, k, sorted=True):
|
|
if any_symbolic_tensors((x,)):
|
|
return TopK().symbolic_call(x, k, sorted)
|
|
return backend.math.top_k(x, k, sorted)
|
|
|
|
|
|
class InTopK(Operation):
|
|
def call(self, targets, predictions, k):
|
|
return backend.math.in_top_k(targets, predictions, k)
|
|
|
|
|
|
def in_top_k(targets, predictions, k):
|
|
if any_symbolic_tensors((targets, predictions)):
|
|
return InTopK().symbolic_call(targets, predictions, k)
|
|
return backend.math.in_top_k(targets, predictions, k)
|