Commit c8692a2a authored by joegomes's avatar joegomes
Browse files

Fixed _standardize_df for non-user defined features. Removed print statements from deep.py

parent f24cc8c6
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -192,6 +192,7 @@ class DataFeaturizer(object):
    df["smiles"] = ori_df[[self.smiles_field]]
    for task in self.tasks:
      df[task] = ori_df[[task]]
    if self.split_field is not None:
      for feature in self.user_specified_features:
        df[feature] = ori_df[[feature]]
    if self.split_field is not None:
+0 −4
Original line number Diff line number Diff line
@@ -150,10 +150,6 @@ class MultiTaskDNN(KerasModel):
        # output the most likely class.
        y_pred_task = np.squeeze(np.argmax(y_pred_dict[taskname], axis=1))
      else:
        print("taskname")
        print(taskname)
        print("y_pred_dict.keys()")
        print(y_pred_dict.keys())
        y_pred_task = np.squeeze(y_pred_dict[taskname])
      y_pred[:, ind] = y_pred_task
    y_pred = np.squeeze(y_pred)