Commit 7d1f4c85 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent 736614c4

deepchem/models/IRV.py

+11 −0
Original line number Diff line number Diff line
@@ -124,3 +124,14 @@ class TensorflowMultitaskIRVClassifier(KerasModel): 
        SigmoidCrossEntropy(),

        output_types=['prediction', 'loss'],

        **kwargs)





class TensorflowMultitaskIRVClassifier(MultitaskIRVClassifier):



  def __init__(self, *args, **kwargs):



    warnings.warn(

        "TensorflowMultitaskIRVClassifier is deprecated and has been renamed to MultitaskIRVClassifier",

        FutureWarning)



    super(TensorflowMultitaskIRVClassifier, self).__init__(*args, **kwargs)

deepchem/models/__init__.py

+2 −1
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@ from deepchem.models.callbacks import ValidationCallback 
from deepchem.models.fcnet import MultitaskRegressor

from deepchem.models.fcnet import MultitaskClassifier

from deepchem.models.fcnet import MultitaskFitTransformRegressor

from deepchem.models.IRV import TensorflowMultitaskIRVClassifier

from deepchem.models.IRV import MultitaskIRVClassifier

from deepchem.models.robust_multitask import RobustMultitaskClassifier

from deepchem.models.robust_multitask import RobustMultitaskRegressor

from deepchem.models.progressive_multitask import ProgressiveMultitaskRegressor, ProgressiveMultitaskClassifier
@@ -29,3 +29,4 @@ from deepchem.models.chemnet_models import Smiles2Vec, ChemCeption 


from deepchem.models.text_cnn import TextCNNTensorGraph

from deepchem.models.graph_models import WeaveTensorGraph, DTNNTensorGraph, DAGTensorGraph, GraphConvTensorGraph, MPNNTensorGraph

from deepchem.models.IRV import TensorflowMultitaskIRVClassifier

CRA Git | Maintained and supported by SUSTech CRA and CCSE