FLParameter

View Source On Gitee

import com.mindspore.flclient.FLParameter

FLParameter is used to define parameters related to federated learning.

Public Member Functions

function

public static synchronized FLParameter getInstance()

public String getDeployEnv()

public void setDeployEnv(String env)

public String getDomainName()

public void setDomainName(String domainName)

public String getClientID()

public void setClientID(String clientID)

public String getCertPath()

public void setCertPath(String certPath)

public SSLSocketFactory getSslSocketFactory()

public void setSslSocketFactory(SSLSocketFactory sslSocketFactory)

public X509TrustManager getX509TrustManager()

public void setX509TrustManager(X509TrustManager x509TrustManager)

public IFLJobResultCallback getIflJobResultCallback()

public void setIflJobResultCallback(IFLJobResultCallback iflJobResultCallback)

public String getFlName()

public void setFlName(String flName)

public String getTrainModelPath()

public void setTrainModelPath(String trainModelPath)

public String getInferModelPath()

public void setInferModelPath(String inferModelPath)

public String getSslProtocol()

public void setSslProtocol(String sslProtocol)

public int getTimeOut()

public void setTimeOut(int timeOut)

public int getSleepTime()

public void setSleepTime(int sleepTime)

public boolean isUseElb()

public void setUseElb(boolean useElb)

public int getServerNum()

public void setServerNum(int serverNum)

public boolean isPkiVerify()

public void setPkiVerify(boolean ifPkiVerify)

public String getEquipCrlPath()

public void setEquipCrlPath(String certPath)

public long getValidInterval()

public void setValidInterval(long validInterval)

public int getThreadNum()

public void setThreadNum(int threadNum)

public int getCpuBindMode()

public void setCpuBindMode(BindMode cpuBindMode)

public List getHybridWeightName(RunType runType)

public void setHybridWeightName(List hybridWeightName, RunType runType)

public Map<RunType, List> getDataMap()

public void setDataMap(Map<RunType, List/> dataMap)

public ServerMod getServerMod()

public void setServerMod(ServerMod serverMod)

public int getBatchSize()

public void setBatchSize(int batchSize)

getInstance

public static synchronized FLParameter getInstance()

Obtains a single FLParameter instance.

  • Return value

    Single object of the FLParameter type.

getDeployEnv

public String getDeployEnv()

Obtains the deployment environment for federated learning set by users.

  • Return value

    The deployment environment for federated learning of the string type.

setDeployEnv

public void setDeployEnv(String env)

Used to set the deployment environment for federated learning, a whitelist is set, currently only “x86”, “android” are supported.

  • Parameter

    • env: the deployment environment for federated learning.

getDomainName

public String getDomainName()

Obtains the domain name set by a user.

  • Return value

    Domain name of the string type.

setDomainName

public void setDomainName(String domainName)

Used to set the url for device-cloud communication. Currently, https and http communication are supported, the corresponding formats are like: https://……, http://……, and when useElb is set to true, the format must be: https://127.0.0.0 : 6666 or http://127.0.0.0 : 6666 , where 127.0.0.0 corresponds to the ip of the machine providing cloud-side services (corresponding to the cloud-side parameter --scheduler_ip), and 6666 corresponds to the cloud-side parameter --fl_server_port.

  • Parameter

    • domainName: domain name.

getClientID

public String getClientID()

A clientID that uniquely identifies the client is automatically generated before each federated learning task is started (if the user needs to set the clientID, setClientID is used to set before the federated learning training task is started), and this method is used to obtain the ID, which can be used to generate the relevant certificates in the device-cloud security authentication scenarios.

  • Return value

    Unique ID of the client, which is of the string type.

setClientID

public void setClientID(String clientID)

Used to set the clientID that uniquely identifies the client.

  • Parameter

    • clientID: unique ID of the client.

getCertPath

public String getCertPath()

Obtains the self-signed root certificate path used for device-cloud HTTPS communication.

  • Return value

    The self-signed root certificate path of the string type.

setCertPath

public void setCertPath(String certPath)

Sets the self-signed root certificate path used for device-cloud HTTPS communication. When the deployment environment is “x86” and the device-cloud uses a self-signed certificate for HTTPS communication authentication, this parameter needs to be set. The certificate must be consistent with the CA root certificate used to generate the cloud-side self-signed certificate to pass the verification. This parameter is used for non-Android scenarios.

  • Parameter

    • certPath: the self-signed root certificate path used for device-cloud HTTPS communication.

getSslSocketFactory

public SSLSocketFactory getSslSocketFactory()

Obtains the ssl certificate authentication library sslSocketFactory set by the user.

  • Return value

    The ssl certificate authentication library sslSocketFactory , which is of the SSLSocketFactory type.

setSslSocketFactory

public void setSslSocketFactory(SSLSocketFactory sslSocketFactory)

Used to set the ssl certificate authentication library sslSocketFactory.

  • Parameter

    • sslSocketFactory: the ssl certificate authentication library.

getX509TrustManager

public X509TrustManager getX509TrustManager()

Obtains the ssl certificate authentication manager x509TrustManager set by the user.

  • Return value

    the ssl certificate authentication manager x509TrustManager, which is of the X509TrustManager type.

setX509TrustManager

public void setX509TrustManager(X509TrustManager x509TrustManager)

Used to set the ssl certificate authentication manager x509TrustManager.

  • Parameter

    • x509TrustManager: the ssl certificate authentication manager.

getIflJobResultCallback

public IFLJobResultCallback getIflJobResultCallback()

Obtains the federated learning callback function object iflJobResultCallback set by the user.

  • Return value

    The federated learning callback function object iflJobResultCallback, which is of the IFLJobResultCallback type.

setIflJobResultCallback

public void setIflJobResultCallback(IFLJobResultCallback iflJobResultCallback)

Used to set the federated learning callback function object iflJobResultCallback, the user can implement the specific method of the interface class IFLJobResultCallback.java in the project according to the needs of the actual scene, and set it as a callback function object in the federated learning task.

  • Parameter

    • iflJobResultCallback: the federated learning callback function object.

getFlName

public String getFlName()

Obtains the package path of model script set by a user.

  • Return value

    Name of the package path of model script of the string type.

setFlName

public void setFlName(String flName)

Sets the package path of model script . We provide two types of model scripts for your reference (Supervised sentiment classification task, Lenet image classification task). For supervised sentiment classification tasks, this parameter can be set to the package path of the provided script file AlBertClient.java, like as com.mindspore.flclient.demo.albert.AlbertClient; for Lenet image classification tasks, this parameter can be set to the package path of the provided script file LenetClient.java, like as com.mindspore.flclient.demo.lenet.LenetClient. At the same time, users can refer to these two types of model scripts, define the model script by themselves, and then set the parameter to the package path of the customized model file ModelClient.java (which needs to inherit from the class Client.java).

  • Parameter

    • flName: package path of model script.

getTrainModelPath

public String getTrainModelPath()

Obtains the path of the training model set by a user.

  • Return value

    Path of the training model of the string type.

setTrainModelPath

public void setTrainModelPath(String trainModelPath)

Sets the path of the training model.

  • Parameter

    • trainModelPath: training model path.

getInferModelPath

public String getInferModelPath()

Obtains the path of the inference model set by a user.

  • Return value

    Path of the inference model of the string type.

setInferModelPath

public void setInferModelPath(String inferModelPath)

Sets the path of the inference model.

  • Parameter

    • inferModelPath: path of the inference model.

getSslProtocol

public String getSslProtocol()

Obtains the TLS protocol version used by the device-cloud HTTPS communication.

  • Return value

    The TLS protocol version used by the device-cloud HTTPS communication of the string type.

setSslProtocol

public void setSslProtocol(String sslProtocol)

Used to set the TLS protocol version used by the device-cloud HTTPS communication, a whitelist is set, and currently only “TLSv1.3” or “TLSv1.2” is supported. Only need to set it up in the HTTPS communication scenario.

  • Parameter

    • sslProtocol: the TLS protocol version used by the device-cloud HTTPS communication.

getTimeOut

public int getTimeOut()

Obtains the timeout interval set by a user for device-side communication.

  • Return value

    Timeout interval for communication on the device, which is an integer.

setTimeOut

public void setTimeOut(int timeOut)

Sets the timeout interval for communication on the device.

  • Parameter

    • timeOut: timeout interval for communication on the device.

getSleepTime

public int getSleepTime()

Obtains the waiting time of repeated requests set by a user.

  • Return value

    Waiting time of repeated requests, which is an integer.

setSleepTime

public void setSleepTime(int sleepTime)

Sets the waiting time of repeated requests.

  • Parameter

    • sleepTime: waiting time for repeated requests.

isUseElb

public boolean isUseElb()

Determines whether the elastic load balancing is simulated, that is, whether a client randomly sends requests to a server address within a specified range.

  • Return value

    The value is of the boolean type. The value true indicates that the client sends requests to a random server address within a specified range. The value false indicates that the client sends a request to a fixed server address.

setUseElb

public void setUseElb(boolean useElb)

Determines whether to simulate the elastic load balancing, that is, whether a client randomly sends a request to a server address within a specified range.

  • Parameter

    • useElb: determines whether to simulate the elastic load balancing. The default value is false.

getServerNum

public int getServerNum()

Obtains the number of servers that can send requests when simulating the elastic load balancing.

  • Return value

    Number of servers that can send requests during elastic load balancing simulation, which is an integer.

setServerNum

public void setServerNum(int serverNum)

Sets the number of servers that can send requests during elastic load balancing simulation.

  • Parameter

    • serverNum: number of servers that can send requests during elastic load balancing simulation. The default value is 1.

isPkiVerify

public boolean isPkiVerify()

Whether to perform device-cloud security authentication.

  • Return value

    The value is of the boolean type. The value true indicates that device-cloud security authentication is performed, and the value false indicates that device-cloud security authentication is not performed.

setPkiVerify

public void setPkiVerify(boolean pkiVerify)

Determines whether to perform device-cloud security authentication.

  • Parameter

    • pkiVerify: whether to perform device-cloud security authentication.

getEquipCrlPath

public String getEquipCrlPath()

Obtains the CRL certification path equipCrlPath of the device certificate set by the user. This parameter is used in the Android environment.

  • Return value

    The certification path of the string type.

setEquipCrlPath

public void setEquipCrlPath(String certPath)

Used to set the CRL certification path of the device certificate. It is used to verify whether the digital certificate is revoked. This parameter is used in the Android environment.

  • Parameter

    • certPath: the certification path.

getValidInterval

public long getValidInterval()

Obtains the valid iteration interval validIterInterval set by the user. This parameter is used in the Android environment.

  • Return value

    The valid iteration interval validIterInterval of the long type.

setValidInterval

public void setValidInterval(long validInterval)

Used to set the valid iteration interval validIterInterval. The recommended duration is the duration of one training epoch between the device-cloud(unit: milliseconds). It is used to prevent replay attacks. This parameter is used in the Android environment.

  • Parameter

    • validInterval: the valid iteration interval validIterInterval.

getThreadNum

public int getThreadNum()

Obtains the number of threads used in federated learning training and inference. The default value is 1.

  • Return value

    The number of threads used in federated learning training and inference, which is of the int type.

setThreadNum

public void setThreadNum(int threadNum)

Used to set the number of threads used in federated learning training and inference.

  • Parameter

    • threadNum: the number of threads used in federated learning training and inference.

getCpuBindMode

public int getCpuBindMode()

Obtains the cpu core that threads need to bind during federated learning training and inference.

  • Return value

    Convert the enumerated type of cpu core to int type and return.

setCpuBindMode

public void setCpuBindMode(BindMode cpuBindMode)

Used to set the cpu core that threads need to bind during federated learning training and inference.

  • Parameter

    • cpuBindMode: it is the enumeration type BindMode, where BindMode.NOT_BINDING_CORE represents the unbound core, which is automatically assigned by the system, BindMode.BIND_LARGE_CORE represents the bound large core, and BindMode.BIND_MIDDLE_CORE represents the bound middle core.

getHybridWeightName

public List<String> getHybridWeightName(RunType runType)

Used in hybrid training mode. Get the training weight name and inference weight name set by the user.

  • Parameter

  • runType: RunType enumeration type, only supports to be set to RunType.TRAINMODE (representing the training weight name) , RunType.INFERMODE (representing the inference weight name).

  • Return value

    A list of corresponding weight names according to the parameter runType, which is of the List type.

setHybridWeightName

public void setHybridWeightName(List<String> hybridWeightName, RunType runType)

Due to the hybrid training mode, part of the weights delivered by the server is imported into the training model, and part is imported into the inference model, but the framework itself cannot judge it, so the user needs to set the relevant training weight name and inference weight name by himself. This method is provided for the user to set.

  • Parameter

    • hybridWeightName: a list of weight names of the List type.

    • runType: RunType enumeration type, only supports setting to RunType.TRAINMODE (representing setting training weight name), RunType.INFERMODE (representing setting reasoning weight name).

getDataMap

public Map<RunType, List<String>> getDataMap()

Obtains the federated learning dataset set by the user.

  • Return value

    the federated learning dataset set of the Map<RunType, List> type.

setDataMap

public void setDataMap(Map<RunType, List<String>> dataMap)

Used to set the federated learning dataset set by the user.

  • Parameter

    • dataMap: the dataset of Map<RunType, List> type, the key in the map is the RunType enumeration type, the value is the corresponding dataset list, when the key is RunType.TRAINMODE, the corresponding value is the training-related dataset list, when the key is RunType.EVALMODE, it means that the corresponding value is a list of verification-related datasets, and when the key is RunType.INFERMODE, it means that the corresponding value is a list of inference-related datasets.

getServerMod

public ServerMod getServerMod()

Obtains the federated learning training mode.

  • Return value

    The federated learning training mode of ServerMod enumeration type.

setServerMod

public void setServerMod(ServerMod serverMod)

Used to set the federated learning training mode.

  • Parameter

    • serverMod: the federated learning training mode of ServerMod enumeration type, where ServerMod.FEDERATED_LEARNING represents the normal federated learning mode (training and inference use the same model) ServerMod.HYBRID_TRAINING represents the hybrid learning mode (training and inference use different models, and the server side also includes training process).

getBatchSize

public int getBatchSize()

Obtains the number of single-step training samples used in federated learning training and inference, that is, batch size.

  • Return value

    BatchSize, the number of single-step training samples of int type.

setBatchSize

public void setBatchSize(int batchSize)

Used to set the number of single-step training samples used in federated learning training and inference, that is, batch size. It needs to be consistent with the batch size of the input data of the model.

  • Parameter

    • batchSize: the number of single-step training samples of int type.