unimore/bda-6-steffo
unimore/bda-6-steffo
1
Fork 0
mirror of https://github.com/Steffo99/unimore-bda-6.git synced 2024-11-25 01:04:19 +00:00
Code Activity

Normalize mean value of tensorflow models

Browse source
This commit is contained in:
Steffo 2023-02-10 17:33:54 +01:00
parent 07be21e809
commit ade3a6bdc7
Signed by: steffo
GPG key ID: 2A24051445686895
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
unimore_bda_6/analysis/tf_text.py
View file

@ -259,7 +259,7 @@ class TensorflowPolarSentimentAnalyzer(TensorflowSentimentAnalyzer):
return model
def _translate_prediction(self, a: numpy.array) -> Category:
return a[0, 0] * 5
return (a[0, 0] + 0.5) * 5
__all__ = (

2
unimore_bda_6/database/datatypes.py
View file

@ -41,7 +41,7 @@ class Review:
return tensorflow.convert_to_tensor(self.text, dtype=tensorflow.string)
def to_tensor_normvalue(self) -> tensorflow.Tensor:
return tensorflow.convert_to_tensor([self.category / 5], dtype=tensorflow.float32)
return tensorflow.convert_to_tensor([self.category / 5 - 0.5], dtype=tensorflow.float32)
def to_tensor_tuple_normvalue(self) -> tuple[tensorflow.Tensor, tensorflow.Tensor]:
return (