Python TorchModel.fit示例

示例#1

 def testTorchModelFit(self):
     torch_model = TorchModel()
     torch_model.fit(
         Xs=[np.array(0)],
         Ys=[np.array(0)],
         Yvars=[np.array(1)],
         bounds=[(0, 1)],
         task_features=[],
         feature_names=["x1"],
     )

示例#2

文件： test_torch.py 项目： proteanblank/Ax

 def testTorchModelFit(self):
     torch_model = TorchModel()
     torch_model.fit(
         Xs=[np.array(0)],
         Ys=[np.array(0)],
         Yvars=[np.array(1)],
         search_space_digest=SearchSpaceDigest(
             feature_names=["x1"],
             bounds=[(0, 1)],
         ),
         metric_names=["y"],
     )