Commit 8b501577 authored by Julian Zimmermann's avatar Julian Zimmermann
Browse files

minor typos

parent 05a4826f
......@@ -26,7 +26,6 @@ FLAGS = flags.FLAGS
flags.DEFINE_float(
'learning_rate', 0.1,
'Base learning rate -> Final lr is multipled by sqrt(batch_size). '
'Rule of thump for efficientnets: ~ 0.016'
'Used in simclr paper: 0.1')
flags.DEFINE_enum(
......
......@@ -95,7 +95,7 @@ def w2_multi_normal(us1, us2):
def lin_acc(embeddings, labels):
"""
This trains a linear classifier and returns the accuracy, precision,
recall and all confusion matrix elements
recall and f1 scores
:param embeddings: The trained embeddings ... This is X
:param labels: The target ... this is Y
:return: Multiple scalars
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment