Commit 11b29d8e authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Change

parent f521cf0e

deepchem/models/IRV.py

+0 −11
Original line number Diff line number Diff line
@@ -124,14 +124,3 @@ class MultitaskIRVClassifier(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

+1 −2
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 MultitaskIRVClassifier

from deepchem.models.IRV import TensorflowMultitaskIRVClassifier

from deepchem.models.robust_multitask import RobustMultitaskClassifier

from deepchem.models.robust_multitask import RobustMultitaskRegressor

from deepchem.models.progressive_multitask import ProgressiveMultitaskRegressor, ProgressiveMultitaskClassifier
@@ -29,4 +29,3 @@ 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

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Diff line number Diff line
@@ -429,7 +429,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    dataset_trans = IRV_transformer.transform(dataset)

    classification_metric = dc.metrics.Metric(

        dc.metrics.accuracy_score, task_averager=np.mean)

    model = dc.models.MultitaskIRVClassifier(

    model = dc.models.TensorflowMultitaskIRVClassifier(

        n_tasks, K=5, learning_rate=0.01, batch_size=n_samples)



    # Fit trained model

CRA Git | Maintained and supported by SUSTech CRA and CCSE