Commit 7d333881 authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

fix for adanet, using tf layers instead of keras one

parent 470638c0
......@@ -349,7 +349,7 @@ class AdaNetWrapper(object):
@staticmethod
def predict_online(h5_file, split, predict_fn=None,
mask_fn=None, batch_size=10000, limit=None,
mask_fn=None, batch_size=1000, limit=None,
probs=False, n_classes=None, model_dir=None):
"""Predict on given testset without killing the memory.
......
......@@ -2,8 +2,8 @@ import functools
import adanet
import numpy as np
import tensorflow.compat.v1 as tf
from tensorflow.keras import layers
from tensorflow.keras.layers import Dense, Dropout
from tensorflow.compat.v1 import layers
from tensorflow.compat.v1.layers import Dense, Dropout
from chemicalchecker.util import logged
......
import functools
import adanet
import tensorflow.compat.v1 as tf
from tensorflow.keras import layers
from tensorflow.keras.layers import Dense, Dropout
from tensorflow.compat.v1 import layers
from tensorflow.compat.v1.layers import Dense, Dropout
from chemicalchecker.util import logged
......
Markdown is supported
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