ddi-fw 0.0.134__py3-none-any.whl → 0.0.136__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -75,16 +75,17 @@ class TFModelWrapper(ModelWrapper):
75
75
 
76
76
  # https://github.com/mlflow/mlflow/blob/master/examples/tensorflow/train.py
77
77
 
78
- def predict(self, best_model):
79
- pred = best_model.predict(self.test_data)
78
+ def predict(self):
79
+ pred = self.best_model.predict(self.test_data)
80
80
  return pred
81
81
 
82
82
  def fit_and_evaluate(self):
83
83
 
84
84
  with mlflow.start_run(run_name=self.descriptor, description="***", nested=True) as run:
85
85
  print(run.info.artifact_uri)
86
- best_model, best_model_key = self.fit()
87
- pred = self.predict(best_model)
86
+ best_model, best_model_key =self.fit()
87
+ self.best_model = best_model
88
+ pred = self.predict()
88
89
  logs, metrics = evaluate(
89
90
  actual=self.test_label, pred=pred, info=self.descriptor)
90
91
  metrics.format_float()
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ddi_fw
3
- Version: 0.0.134
3
+ Version: 0.0.136
4
4
  Summary: Do not use :)
5
5
  Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
6
6
  Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
@@ -78,7 +78,7 @@ ddi_fw/ml/evaluation_helper.py,sha256=o4-w5Xa3t4olLW4ymx_8L-Buhe5wfQEmT2bh4Zz544
78
78
  ddi_fw/ml/ml_helper.py,sha256=diDHIzcD0CkdRB3lKu_FFH2frpAUTFfI4xKOAxYA04I,6167
79
79
  ddi_fw/ml/model_wrapper.py,sha256=kc01_TVJuriUvNI6ABnLngnJWvmG_Y7-XJ6XMusLJ8U,1088
80
80
  ddi_fw/ml/pytorch_wrapper.py,sha256=AkG-2sKDXr0IBhgmkbjG0i20OuwQv3mhdvqp6UvJDCA,3716
81
- ddi_fw/ml/tensorflow_wrapper.py,sha256=jaWPZfsQMSG_Vzl1OTALAbgb2sdJklcgQD5kokt7b18,5409
81
+ ddi_fw/ml/tensorflow_wrapper.py,sha256=csNPBnUdP4mZgISFkk-PZdZZBb49sveJwJQkMp2RdIA,5433
82
82
  ddi_fw/ner/__init__.py,sha256=JwhGXrepomxPSsGsg2b_xPRC72AjvxOIn2CW5Mvscn0,26
83
83
  ddi_fw/ner/mmlrestclient.py,sha256=NZta7m2Qm6I_qtVguMZhqtAUjVBmmXn0-TMnsNp0jpg,6859
84
84
  ddi_fw/ner/ner.py,sha256=BEs9AFljAxOQrC2BEP1raSzRoypcfELS5UTdl4bjTqw,15863
@@ -106,7 +106,7 @@ ddi_fw/utils/package_helper.py,sha256=erl8_onmhK-41zQoaED2qyDUV9GQxmT9sdoyRp9_q5
106
106
  ddi_fw/utils/py7zr_helper.py,sha256=gOqaFIyJvTjUM-btO2x9AQ69jZOS8PoKN0wetYIckJw,4747
107
107
  ddi_fw/utils/utils.py,sha256=szwnxMTDRrZoeNRyDuf3aCbtzriwtaRk4mHSH3asLdA,4301
108
108
  ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,5567
109
- ddi_fw-0.0.134.dist-info/METADATA,sha256=EfKejA6cXIrw_TG4WzFg3ui0vXwIiWPJzqhubEENdXA,1965
110
- ddi_fw-0.0.134.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
111
- ddi_fw-0.0.134.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
112
- ddi_fw-0.0.134.dist-info/RECORD,,
109
+ ddi_fw-0.0.136.dist-info/METADATA,sha256=mwHv-vQfu6q78kJAuOhz_snm4XUjIcD-8n_UE6Ey_J8,1965
110
+ ddi_fw-0.0.136.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
111
+ ddi_fw-0.0.136.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
112
+ ddi_fw-0.0.136.dist-info/RECORD,,