mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-09 22:10:56 +08:00
@@ -32,7 +32,7 @@ import abc
|
||||
import enum
|
||||
|
||||
|
||||
# Type defintions
|
||||
# Type definitions
|
||||
class DataTypes(enum.IntEnum):
|
||||
"""Defines numerical types of each column."""
|
||||
|
||||
|
||||
@@ -254,9 +254,9 @@ class DistributedHyperparamOptManager(HyperparamOptManager):
|
||||
param_ranges: Discrete hyperparameter range for random search.
|
||||
fixed_params: Fixed model parameters per experiment.
|
||||
root_model_folder: Folder to store optimisation artifacts.
|
||||
worker_number: Worker index definining which set of hyperparameters to
|
||||
worker_number: Worker index defining which set of hyperparameters to
|
||||
test.
|
||||
search_iterations: Maximum numer of random search iterations.
|
||||
search_iterations: Maximum number of random search iterations.
|
||||
num_iterations_per_worker: How many iterations are handled per worker.
|
||||
clear_serialised_params: Whether to regenerate hyperparameter
|
||||
combinations.
|
||||
@@ -330,7 +330,7 @@ class DistributedHyperparamOptManager(HyperparamOptManager):
|
||||
if os.path.exists(self.serialised_ranges_folder):
|
||||
df = pd.read_csv(self.serialised_ranges_path, index_col=0)
|
||||
else:
|
||||
print("Unable to load - regenerating serach ranges instead")
|
||||
print("Unable to load - regenerating search ranges instead")
|
||||
df = self.update_serialised_hyperparam_df()
|
||||
|
||||
return df
|
||||
|
||||
@@ -342,7 +342,7 @@ class TFTDataCache:
|
||||
|
||||
@classmethod
|
||||
def contains(cls, key):
|
||||
"""Retuns boolean indicating whether key is present in cache."""
|
||||
"""Returns boolean indicating whether key is present in cache."""
|
||||
|
||||
return key in cls._data_cache
|
||||
|
||||
@@ -1120,10 +1120,10 @@ class TemporalFusionTransformer:
|
||||
Args:
|
||||
df: Input dataframe
|
||||
return_targets: Whether to also return outputs aligned with predictions to
|
||||
faciliate evaluation
|
||||
facilitate evaluation
|
||||
|
||||
Returns:
|
||||
Input dataframe or tuple of (input dataframe, algined output dataframe).
|
||||
Input dataframe or tuple of (input dataframe, aligned output dataframe).
|
||||
"""
|
||||
|
||||
data = self._batch_data(df)
|
||||
|
||||
@@ -295,7 +295,7 @@ class TFTModel(ModelFT):
|
||||
def to_pickle(self, path: Union[Path, str]):
|
||||
"""
|
||||
Tensorflow model can't be dumped directly.
|
||||
So the data should be save seperatedly
|
||||
So the data should be save separately
|
||||
|
||||
**TODO**: Please implement the function to load the files
|
||||
|
||||
|
||||
Reference in New Issue
Block a user